diff --git a/bob/bio/base/transformers/algorithm.py b/bob/bio/base/transformers/algorithm.py index 1fe164d0203486dc6a9e40159fd6df72b55ecdac..f49b906c0089ad1fd79e419e6bb2481ecd186508 100644 --- a/bob/bio/base/transformers/algorithm.py +++ b/bob/bio/base/transformers/algorithm.py @@ -75,7 +75,4 @@ class AlgorithmTransformer(TransformerMixin, BaseEstimator): ] def _more_tags(self): - if self.callable.requires_training: - return {"stateless": False, "requires_fit": True} - else: - return {"stateless": True, "requires_fit": False} + return {"stateless": not self.callable.requires_training, "requires_fit": self.callable.requires_training} diff --git a/bob/bio/base/transformers/extractor.py b/bob/bio/base/transformers/extractor.py index e2ddcb327d4efd24984c50070824bf8008737a0d..40778e77442498a0ae4a3246fbed75659354bce2 100644 --- a/bob/bio/base/transformers/extractor.py +++ b/bob/bio/base/transformers/extractor.py @@ -58,7 +58,7 @@ class ExtractorTransformer(TransformerMixin, BaseEstimator): ] def _more_tags(self): - if self.callable.requires_training: - return {"stateless": False, "requires_fit": True} - else: - return {"stateless": True, "requires_fit": False} + return { + "stateless": not self.callable.requires_training, + "requires_fit": self.callable.requires_training, + }