requires_fit is not all available all the time

parent 49eaeff1
Pipeline #38381 failed with stage
in 3 minutes and 32 seconds
......@@ -245,7 +245,7 @@ class SampleMixin(BaseEstimator):
# See:
# if the estimator does not require fit or is stateless don't call fit
tags = self._get_tags()
if tags["stateless"] or not tags["requires_fit"]:
if tags["stateless"] or ("requires_fit" in tags and not tags["requires_fit"]):
return self
# if the estimator needs to be fitted.
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment