diff --git a/bob/bio/base/config/examples/pca_atnt.py b/bob/bio/base/config/examples/pca_atnt.py index 792e6774d5f257ddc914cdddd14cf4a4fd1f3507..afffe2df476045a6509b2443269751b66fae0df9 100644 --- a/bob/bio/base/config/examples/pca_atnt.py +++ b/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) diff --git a/bob/bio/base/pipelines/vanilla_biometrics/mixins.py b/bob/bio/base/pipelines/vanilla_biometrics/mixins.py index a61f9a599200b702fd746d0c070ec44f92135b83..c335a9aae5a4e9f142fd155f44541304aac7da29 100644 --- a/bob/bio/base/pipelines/vanilla_biometrics/mixins.py +++ b/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 diff --git a/bob/bio/base/script/vanilla_biometrics.py b/bob/bio/base/script/vanilla_biometrics.py index 26c9a008b7b742fddce0de42db312d70c38d31b7..abe07934b0b91eea228110e9e8d3ccdfe3c73cda 100644 --- a/bob/bio/base/script/vanilla_biometrics.py +++ b/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)