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

Cleaning up

parent 9e3a1f03
No related branches found
No related tags found
2 merge requests!185Wrappers and aggregators,!180[dask] Preparing bob.bio.base for dask pipelines
......@@ -146,28 +146,3 @@ class VanillaBiometricsPipeline(object):
# scores is a list of Samples
return scores
def dask_vanilla_biometrics(pipeline, npartitions=None):
"""
Given a :py:class:`VanillaBiometrics`, wraps :py:meth:`VanillaBiometrics.transformer` and
:py:class:`VanillaBiometrics.biometric_algorithm` with Dask delayeds
Parameters
----------
pipeline: :py:class:`VanillaBiometrics`
Vanilla Biometrics based pipeline to be dasked
npartitions: int
Number of partitions for the initial `Dask.bag`
"""
from bob.pipelines.mixins import estimator_dask_it, mix_me_up
from bob.bio.base.pipelines.vanilla_biometrics.mixins import BioAlgDaskMixin
transformer = estimator_dask_it(pipeline.transformer, npartitions=npartitions)
biometric_algorithm = mix_me_up([BioAlgDaskMixin], pipeline.biometric_algorithm)
return VanillaBiometrics(transformer, biometric_algorithm)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment