Commit 95624011 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

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.
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