Bug fix: make check_is_fitted call compatible with sklearn v0.21

dynamicslab · Jun 23, 2020 · 8afdc83 · 8afdc83
1 parent 6c97bbc
commit 8afdc83
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pysindy/feature_library/custom_library.py b/pysindy/feature_library/custom_library.py
@@ -89,7 +89,7 @@ def get_feature_names(self, input_features=None):
         output_feature_names : list of string, length n_output_features
 
         """
-        check_is_fitted(self)
+        check_is_fitted(self, "n_output_features_")
         if input_features is None:
             input_features = ["x%d" % i for i in range(self.n_input_features_)]
         feature_names = []
@@ -147,7 +147,7 @@ def transform(self, X):
             generated from applying the custom functions to the inputs.
 
         """
-        check_is_fitted(self)
+        check_is_fitted(self, "n_output_features_")
 
         X = check_array(X)