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

Added kwargs in wrap_bob_legacy

parent 3a991006
No related branches found
No related tags found
2 merge requests!185Wrappers and aggregators,!180[dask] Preparing bob.bio.base for dask pipelines
Pipeline #39672 passed
...@@ -20,6 +20,7 @@ def wrap_bob_legacy( ...@@ -20,6 +20,7 @@ def wrap_bob_legacy(
fit_extra_arguments=(("y", "subject"),), fit_extra_arguments=(("y", "subject"),),
transform_extra_arguments=None, transform_extra_arguments=None,
dask_it=False, dask_it=False,
**kwargs
): ):
""" """
Wraps either :any:`bob.bio.base.preprocessor.Preprocessor`, :any:`bob.bio.base.extractor.Extractor` Wraps either :any:`bob.bio.base.preprocessor.Preprocessor`, :any:`bob.bio.base.extractor.Extractor`
...@@ -50,18 +51,21 @@ def wrap_bob_legacy( ...@@ -50,18 +51,21 @@ def wrap_bob_legacy(
if isinstance(bob_object, Preprocessor): if isinstance(bob_object, Preprocessor):
transformer = wrap_checkpoint_preprocessor( transformer = wrap_checkpoint_preprocessor(
bob_object, features_dir=os.path.join(dir_name, "preprocessor"), bob_object, features_dir=os.path.join(dir_name, "preprocessor"),
**kwargs
) )
elif isinstance(bob_object, Extractor): elif isinstance(bob_object, Extractor):
transformer = wrap_checkpoint_extractor( transformer = wrap_checkpoint_extractor(
bob_object, bob_object,
features_dir=os.path.join(dir_name, "extractor"), features_dir=os.path.join(dir_name, "extractor"),
model_path=dir_name, model_path=dir_name,
**kwargs
) )
elif isinstance(bob_object, Algorithm): elif isinstance(bob_object, Algorithm):
transformer = wrap_checkpoint_algorithm( transformer = wrap_checkpoint_algorithm(
bob_object, bob_object,
features_dir=os.path.join(dir_name, "algorithm"), features_dir=os.path.join(dir_name, "algorithm"),
model_path=dir_name, model_path=dir_name,
**kwargs
) )
else: else:
raise ValueError( raise ValueError(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment