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): ...@@ -245,7 +245,7 @@ class SampleMixin(BaseEstimator):
# See: https://scikit-learn.org/stable/developers/develop.html # See: https://scikit-learn.org/stable/developers/develop.html
# if the estimator does not require fit or is stateless don't call fit # if the estimator does not require fit or is stateless don't call fit
tags = self._get_tags() 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 return self
# if the estimator needs to be fitted. # if the estimator needs to be fitted.
......
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