Skip to content
Snippets Groups Projects

Wrappers and aggregators

Merged Tiago de Freitas Pereira requested to merge wrapper-api into dask-pipelines
1 file
+ 7
4
Compare changes
  • Side-by-side
  • Inline
@@ -191,13 +191,16 @@ def test_checkpoint_bioalg_as_transformer():
@@ -191,13 +191,16 @@ def test_checkpoint_bioalg_as_transformer():
database.probes(),
database.probes(),
allow_scoring_with_all_biometric_references=database.allow_scoring_with_all_biometric_references,
allow_scoring_with_all_biometric_references=database.allow_scoring_with_all_biometric_references,
)
)
if with_dask:
filename = os.path.join(dir_name, "concatenated_scores.txt")
scores = scores.compute(scheduler="single-threaded")
score_writer.concatenate_write_scores(scores, filename)
if isinstance(score_writer, CSVScoreWriter):
if isinstance(score_writer, CSVScoreWriter):
assert len(open(filename).readlines()) == 101
base_path = os.path.join(dir_name, "concatenated_scores")
 
score_writer.concatenate_write_scores(scores, base_path)
 
assert len(open(os.path.join(base_path, "chunk_0.csv")).readlines()) == 101
else:
else:
 
filename = os.path.join(dir_name, "concatenated_scores.txt")
 
score_writer.concatenate_write_scores(scores, filename)
assert len(open(filename).readlines()) == 100
assert len(open(filename).readlines()) == 100
run_pipeline(False)
run_pipeline(False)
Loading