moved fitted_ after function call (#77)

* moved fitted_ after function call * also moved fitted_ after function call in outlier
koaning · Mar 31, 2023 · a928a09 · a928a09
1 parent 97a80ed
commit a928a09
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/hulearn/classification/functionclassifier.py b/hulearn/classification/functionclassifier.py
@@ -45,17 +45,17 @@ def fit(self, X, y):
         Fit the classifier. No-Op.
         """
         # Run it to confirm no error happened.
-        self.fitted_ = True
         _ = self.func(X, **self.kwargs)
+        self.fitted_ = True
         return self
 
     def partial_fit(self, X, y, classes=None, sample_weight=None):
         """
         Fit the classifier partially. No-Op.
         """
         # Run it to confirm no error happened.
-        self.fitted_ = True
         _ = self.func(X, **self.kwargs)
+        self.fitted_ = True
         return self
 
     def predict(self, X):

diff --git a/hulearn/outlier/functionoutlier.py b/hulearn/outlier/functionoutlier.py
@@ -23,8 +23,8 @@ def fit(self, X, y=None):
         Fit the classifier. No-Op.
         """
         # Run it to confirm no error happened.
-        self.fitted_ = True
         _ = self.func(X, **self.kwargs)
+        self.fitted_ = True
         return self
 
     def partial_fit(self, X, y=None):