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):
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
Supports Markdown
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