Commit b2a93711 authored by Yannick DAYER's avatar Yannick DAYER
Browse files

use bob_output in checkpoint wrapper

parent 2668c498
Pipeline #52700 failed with stage
in 10 minutes and 44 seconds
...@@ -292,7 +292,7 @@ class CheckpointWrapper(BaseWrapper, TransformerMixin): ...@@ -292,7 +292,7 @@ class CheckpointWrapper(BaseWrapper, TransformerMixin):
"bob_features_load_fn", bob.io.base.load "bob_features_load_fn", bob.io.base.load
) )
self.sample_attribute = sample_attribute or estimator._get_tags().get( self.sample_attribute = sample_attribute or estimator._get_tags().get(
"bob_checkpoint_attribute", "data" "bob_output", "data"
) )
self.hash_fn = hash_fn self.hash_fn = hash_fn
self.attempts = attempts self.attempts = attempts
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment