fss tag to allow nans

EpistasisLab · Aug 12, 2024 · 7906c44 · 7906c44
1 parent 9b65cb2
commit 7906c44
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tpot2/builtin_modules/feature_set_selector.py b/tpot2/builtin_modules/feature_set_selector.py
@@ -92,6 +92,9 @@ def fit(self, X, y=None):
 
     # def transform(self, X):
 
+    def _get_tags(self):
+        tags = {"allow_nan": True, "requires_y": False}
+        return tags
 
     def _get_support_mask(self):
         """