debug

VIDA-NYU · Feb 20, 2024 · 9e8c32b · 9e8c32b
1 parent 268854d
commit 9e8c32b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/alpha_automl/pipeline_synthesis/setup_search.py b/alpha_automl/pipeline_synthesis/setup_search.py
@@ -171,19 +171,19 @@ def compute_metafeatures(metadata):
         # CATEGORICAL
         metafeatures.append(
             len(nonnumeric_columns["CATEGORICAL_ENCODER"])
-            if "TEXT_ENCODER" in nonnumeric_columns
+            if "CATEGORICAL_ENCODER" in nonnumeric_columns
             else 0
         )
         # DATETIME
         metafeatures.append(
             len(nonnumeric_columns["DATETIME_ENCODER"])
-            if "TEXT_ENCODER" in nonnumeric_columns
+            if "DATETIME_ENCODER" in nonnumeric_columns
             else 0
         )
         # IMAGE
         metafeatures.append(
             len(nonnumeric_columns["IMAGE_ENCODER"])
-            if "TEXT_ENCODER" in nonnumeric_columns
+            if "IMAGE_ENCODER" in nonnumeric_columns
             else 0
         )
     else: