From 0bedccc0c6778b593223f3abbe8afc60b6401769 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Tue, 5 May 2020 10:11:25 +0200
Subject: [PATCH] Added kwargs in wrap_bob_legacy
---
bob/bio/base/wrappers.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/bob/bio/base/wrappers.py b/bob/bio/base/wrappers.py
index b37cb730..8981b002 100644
--- a/bob/bio/base/wrappers.py
+++ b/bob/bio/base/wrappers.py
@@ -20,6 +20,7 @@ def wrap_bob_legacy(
fit_extra_arguments=(("y", "subject"),),
transform_extra_arguments=None,
dask_it=False,
+ **kwargs
):
"""
Wraps either :any:`bob.bio.base.preprocessor.Preprocessor`, :any:`bob.bio.base.extractor.Extractor`
@@ -50,18 +51,21 @@ def wrap_bob_legacy(
if isinstance(bob_object, Preprocessor):
transformer = wrap_checkpoint_preprocessor(
bob_object, features_dir=os.path.join(dir_name, "preprocessor"),
+ **kwargs
)
elif isinstance(bob_object, Extractor):
transformer = wrap_checkpoint_extractor(
bob_object,
features_dir=os.path.join(dir_name, "extractor"),
model_path=dir_name,
+ **kwargs
)
elif isinstance(bob_object, Algorithm):
transformer = wrap_checkpoint_algorithm(
bob_object,
features_dir=os.path.join(dir_name, "algorithm"),
model_path=dir_name,
+ **kwargs
)
else:
raise ValueError(
--
GitLab