Skip to content
Snippets Groups Projects
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
No related branches found
No related tags found
1 merge request!13Updates
Pipeline #38381 failed
......@@ -245,7 +245,7 @@ class SampleMixin(BaseEstimator):
# See: https://scikit-learn.org/stable/developers/develop.html
# 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.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment