Skip to content
Snippets Groups Projects
Commit 9dde3c25 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Clening up checkpoint wrapper

parent a5c3ecef
No related branches found
No related tags found
1 merge request!180[dask] Preparing bob.bio.base for dask pipelines
Pipeline #41506 passed
...@@ -161,6 +161,7 @@ class BioAlgorithmCheckpointWrapper(BioAlgorithm): ...@@ -161,6 +161,7 @@ class BioAlgorithmCheckpointWrapper(BioAlgorithm):
_make_name(sampleset, biometric_references) + self._score_extension, _make_name(sampleset, biometric_references) + self._score_extension,
) )
parent = sampleset
if self.force or not os.path.exists(path): if self.force or not os.path.exists(path):
# Computing score # Computing score
...@@ -170,14 +171,11 @@ class BioAlgorithmCheckpointWrapper(BioAlgorithm): ...@@ -170,14 +171,11 @@ class BioAlgorithmCheckpointWrapper(BioAlgorithm):
allow_scoring_with_all_biometric_references=allow_scoring_with_all_biometric_references, allow_scoring_with_all_biometric_references=allow_scoring_with_all_biometric_references,
) )
self.write_scores(scored_sample_set.samples, path) self.write_scores(scored_sample_set.samples, path)
parent = scored_sample_set
scored_sample_set = DelayedSampleSet( scored_sample_set = DelayedSampleSet(
functools.partial(_load, path), parent=scored_sample_set functools.partial(_load, path), parent=parent
) )
else:
samples = _load(path)
scored_sample_set = SampleSet(samples, parent=sampleset)
return scored_sample_set return scored_sample_set
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment