use bob_output in checkpoint wrapper

b2a93711 · Yannick DAYER · 2668c498 · b2a93711
Commit b2a93711 authored Jul 26, 2021 by Yannick DAYER
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

bob/pipelines/wrappers.py bob/pipelines/wrappers.py +1 -1

No files found.
--- a/bob/pipelines/wrappers.py
+++ b/bob/pipelines/wrappers.py
@@ -292,7 +292,7 @@ class CheckpointWrapper(BaseWrapper, TransformerMixin):
            "bob_features_load_fn", bob.io.base.load
        )
        self.sample_attribute = sample_attribute or estimator._get_tags().get(
-            "bob_checkpoint_attribute", "data"
+            "bob_output", "data"
        )
        self.hash_fn = hash_fn
        self.attempts = attempts