From afbe3b9ecf2ea9e03baf3cad6db2e94825b4309f Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Sat, 2 May 2020 17:10:02 +0200
Subject: [PATCH] Using single-threaded dask scheduler
---
bob/bio/base/test/test_transformers.py | 2 +-
bob/bio/base/test/test_vanilla_biometrics.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/bob/bio/base/test/test_transformers.py b/bob/bio/base/test/test_transformers.py
index ed8575b8..e78049d7 100644
--- a/bob/bio/base/test/test_transformers.py
+++ b/bob/bio/base/test/test_transformers.py
@@ -277,7 +277,7 @@ def test_wrap_bob_pipeline():
if with_dask:
pipeline = mario.wrap(["dask"], pipeline)
transformed_samples = (
- pipeline.fit(training_samples).transform(test_samples).compute()
+ pipeline.fit(training_samples).transform(test_samples).compute(scheduler="single-threaded")
)
else:
transformed_samples = pipeline.fit(training_samples).transform(
diff --git a/bob/bio/base/test/test_vanilla_biometrics.py b/bob/bio/base/test/test_vanilla_biometrics.py
index 2a85d334..99920add 100644
--- a/bob/bio/base/test/test_vanilla_biometrics.py
+++ b/bob/bio/base/test/test_vanilla_biometrics.py
@@ -148,7 +148,7 @@ def test_on_memory():
)
if with_dask:
- scores = scores.compute()
+ scores = scores.compute(scheduler="single-threaded")
assert len(scores) == 10
for probe_ssets in scores:
--
GitLab