[flake]

parent b6994dd0
Pipeline #50457 failed with stage
in 3 minutes and 40 seconds
......@@ -141,7 +141,8 @@ class SampleWrapper(BaseWrapper, TransformerMixin):
if isinstance(samples[0], SampleSet):
return [
SampleSet(
self._samples_transform(sset.samples, method_name), parent=sset,
self._samples_transform(sset.samples, method_name),
parent=sset,
)
for sset in samples
]
......@@ -381,9 +382,9 @@ class CheckpointWrapper(BaseWrapper, TransformerMixin):
self.save_func(to_save, path)
# test loading
loaded = self.load_func(path)
self.load_func(path)
break
except:
except Exception as e:
pass
else:
raise RuntimeError(f"Could not save {to_save} doing {self}.save")
......@@ -437,7 +438,11 @@ class DaskWrapper(BaseWrapper, TransformerMixin):
"""
def __init__(
self, estimator, fit_tag=None, transform_tag=None, **kwargs,
self,
estimator,
fit_tag=None,
transform_tag=None,
**kwargs,
):
super().__init__(**kwargs)
self.estimator = estimator
......
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