upgini 1.1.283__tar.gz → 1.1.284__tar.gz

{upgini-1.1.283 → upgini-1.1.284}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.1.283
+Version: 1.1.284
 Summary: Intelligent data search & enrichment for Machine Learning
 Project-URL: Bug Reports, https://github.com/upgini/upgini/issues
 Project-URL: Homepage, https://upgini.com/

upgini-1.1.284/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.1.284"

{upgini-1.1.283 → upgini-1.1.284}/src/upgini/metrics.py

@@ -549,14 +549,21 @@ class CatBoostWrapper(EstimatorWrapper):
         except Exception as e:
             if "Dictionary size is 0" in e.args[0] and self.text_features:
                 high_cardinality_features = FeaturesValidator.find_high_cardinality(x[self.text_features])
-                self.logger.warning(
-                    "Calculate metrics has problem with CatBoost text features. Try to remove high cardinality"
-                    f" text features {high_cardinality_features} and retry"
-                )
+                if len(high_cardinality_features) == 0:
+                    high_cardinality_features = self.text_features
+                    self.logger.warning(
+                        "Calculate metrics has problem with CatBoost text features. High cardinality features not found"
+                        f". Try to remove all text features {high_cardinality_features} and retry"
+                    )
+                else:
+                    self.logger.warning(
+                        "Calculate metrics has problem with CatBoost text features. Try to remove high cardinality"
+                        f" text features {high_cardinality_features} and retry"
+                    )
                 for f in high_cardinality_features:
                     self.text_features.remove(f)
                     self.exclude_features.append(f)
-                    x = x.drop(columns=f)
+                    x = x.drop(columns=f, errors="ignore")
                 return super().cross_val_predict(x, y, baseline_score_column)
             else:
                 raise e

upgini-1.1.283/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.1.283"