Making dask work again

bob/bio/base/config/examples/pca_atnt.py

@@ -24,5 +24,5 @@ from bob.bio.base.pipelines.vanilla_biometrics.mixins import (
     BioAlgDaskMixin,
 )
 
-#transformer = estimator_dask_it(transformer)
-#algorithm = mix_me_up(BioAlgDaskMixin, algorithm)
+transformer = estimator_dask_it(transformer)
+algorithm = mix_me_up(BioAlgDaskMixin, algorithm)

bob/bio/base/pipelines/vanilla_biometrics/mixins.py

@@ -87,7 +87,7 @@ class BioAlgCheckpointMixin(CheckpointMixin):
 class BioAlgDaskMixin:
     def enroll_samples(self, biometric_reference_features):
         biometric_references = biometric_reference_features.map_partitions(
-            self.enroll_samples
+            super().enroll_samples
         )
         return biometric_references
 
@@ -102,5 +102,5 @@ class BioAlgDaskMixin:
 
         all_references = dask.delayed(list)(biometric_references)
 
-        scores = probe_features.map_partitions(self.score_samples, all_references)
+        scores = probe_features.map_partitions(super().score_samples, all_references)
         return scores

bob/bio/base/script/vanilla_biometrics.py

@@ -183,7 +183,7 @@ def vanilla_biometrics(
                     logger.warning(
                         "`dask_client` not set. Your pipeline will run locally"
                     )
-                    result = result.compute()
+                    result = result.compute(scheduler="single-threaded")
 
             # Flatting out the list
             result = itertools.chain(*result)