Skip to content

Commit

Permalink
Bug fix: make check_is_fitted call compatible with sklearn v0.21
Browse files Browse the repository at this point in the history
  • Loading branch information
briandesilva committed Jun 23, 2020
1 parent 6c97bbc commit 8afdc83
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions pysindy/feature_library/custom_library.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 = []
Expand Down Expand Up @@ -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)

Expand Down

0 comments on commit 8afdc83

Please sign in to comment.