upgini 1.2.9a2__tar.gz → 1.2.9a99__tar.gz

{upgini-1.2.9a2 → upgini-1.2.9a99}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.9a2
+Version: 1.2.9a99
 Summary: Intelligent data search & enrichment for Machine Learning
 Project-URL: Bug Reports, https://github.com/upgini/upgini/issues
 Project-URL: Homepage, https://upgini.com/
@@ -28,6 +28,7 @@ Requires-Dist: fastparquet>=0.8.1
 Requires-Dist: ipywidgets>=8.1.0
 Requires-Dist: jarowinkler>=2.0.0
 Requires-Dist: levenshtein>=0.25.1
+Requires-Dist: lightgbm>=3.3.2
 Requires-Dist: numpy<=1.26.4,>=1.19.0
 Requires-Dist: pandas<3.0.0,>=1.1.0
 Requires-Dist: pydantic<3.0.0,>1.0.0

{upgini-1.2.9a2 → upgini-1.2.9a99}/pyproject.toml

@@ -38,7 +38,7 @@ dependencies = [
     "catboost>=1.0.3",
     "fastparquet>=0.8.1",
     "ipywidgets>=8.1.0",
-
+    "lightgbm>=3.3.2",
     "numpy>=1.19.0,<=1.26.4",
     "pandas>=1.1.0,<3.0.0",
     "pydantic>1.0.0,<3.0.0",

upgini-1.2.9a99/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.2.9a99"

{upgini-1.2.9a2 → upgini-1.2.9a99}/src/upgini/__init__.py

@@ -1,7 +1,7 @@
 import os
 
-# from upgini.features_enricher import FeaturesEnricher  # noqa: F401
-# from upgini.metadata import SearchKey, CVType, RuntimeParameters, ModelTaskType  # noqa: F401
+from upgini.features_enricher import FeaturesEnricher  # noqa: F401
+from upgini.metadata import SearchKey, CVType, RuntimeParameters, ModelTaskType  # noqa: F401
 # from .lazy_import import LazyImport
 
 os.environ["SETUPTOOLS_USE_DISTUTILS"] = "stdlib"

{upgini-1.2.9a2 → upgini-1.2.9a99}/src/upgini/metrics.py

@@ -10,6 +10,7 @@ import catboost
 import numpy as np
 import pandas as pd
 from catboost import CatBoostClassifier, CatBoostRegressor
+from lightgbm import LGBMClassifier, LGBMRegressor
 from numpy import log1p
 from pandas.api.types import is_numeric_dtype
 from sklearn.metrics import check_scoring, get_scorer, make_scorer, roc_auc_score
@@ -405,7 +406,6 @@ class EstimatorWrapper:
                 estimator = CatBoostWrapper(**kwargs)
             else:
                 try:
-                    from lightgbm import LGBMClassifier, LGBMRegressor
                     if isinstance(estimator, (LGBMClassifier, LGBMRegressor)):
                         estimator = LightGBMWrapper(**kwargs)
                     else:
@@ -486,7 +486,7 @@ class CatBoostWrapper(EstimatorWrapper):
 
         # Find rest categorical features
         self.cat_features = _get_cat_features(x, self.text_features, embedding_features)
-        # x = fill_na_cat_features(x, self.cat_features)
+        x = fill_na_cat_features(x, self.cat_features)  # Was commented
         unique_cat_features = []
         for name in self.cat_features:
             # Remove constant categorical features
@@ -536,7 +536,7 @@ class CatBoostWrapper(EstimatorWrapper):
             x, emb_columns = self.group_embeddings(x)
             params["embedding_features"] = emb_columns
         if self.cat_features:
-            # x = fill_na_cat_features(x, self.cat_features)
+            x = fill_na_cat_features(x, self.cat_features)  # Was commented
             params["cat_features"] = self.cat_features
 
         return x, y, params

upgini-1.2.9a2/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.2.9a2"