upgini 1.2.86a1__tar.gz → 1.2.86.dev1__tar.gz

{upgini-1.2.86a1 → upgini-1.2.86.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.86a1
+Version: 1.2.86.dev1
 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.2.86a1 → upgini-1.2.86.dev1}/pyproject.toml

@@ -91,11 +91,7 @@ test_all = 'pytest -s -vv tests'
 
 [[tool.hatch.envs.test.matrix]]
 python = ["3.10", "3.11"]
-pandas = ["1.2.0", "1.3.0", "1.4.0", "1.5.0", "2.0.0"]
-
-[[tool.hatch.envs.test.matrix]]
-python = ["3.10", "3.11"]
-pandas = ["2.1.0", "2.2.0"]
+pandas = ["1.2.0", "1.3.0", "1.4.0", "1.5.0", "2.0.0", "2.1.0", "2.2.0"]
 
 # from versions: 0.1, 0.2, 0.3.0, 0.4.0, 0.4.1, 0.4.2, 0.4.3, 0.5.0, 0.6.0, 0.6.1, 0.7.0, 0.7.1, 0.7.2, 0.7.3, 0.8.0, 0.8.1, 0.9.0, 0.9.1, 0.10.0, 0.10.1, 0.11.0, 0.12.0, 0.13.0, 0.13.1, 0.14.0, 0.14.1, 0.15.0, 0.15.1, 0.15.2, 0.16.0, 0.16.1, 0.16.2, 0.17.0, 0.17.1, 0.18.0, 0.18.1, 0.19.0, 0.19.1, 0.19.2, 0.20.0, 0.20.1, 0.20.2, 0.20.3, 0.21.0, 0.21.1, 0.22.0, 0.23.0, 0.23.1, 0.23.2, 0.23.3, 0.23.4, 0.24.0, 0.24.1, 0.24.2, 0.25.0, 0.25.1, 0.25.2, 0.25.3, 1.0.0, 1.0.1, 1.0.2, 1.0.3, 1.0.4, 1.0.5, 1.1.0, 1.1.1, 1.1.2, 1.1.3, 1.1.4, 1.1.5, 1.2.0, 1.2.1, 1.2.2, 1.2.3, 1.2.4, 1.2.5, 1.3.0, 1.3.1, 1.3.2, 1.3.3, 1.3.4, 1.3.5, 1.4.0rc0, 1.4.0, 1.4.1, 1.4.2, 1.4.3, 1.4.4, 1.5.0rc0, 1.5.0, 1.5.1, 1.5.2, 1.5.3, 2.0.0rc0, 2.0.0rc1, 2.0.0, 2.0.1, 2.0.2, 2.0.3
 
@@ -105,7 +101,6 @@ dependencies = [
   "coverage[toml]",
   "pytest",
   "pytest-cov",
-#  "pytest-timeout",
   "requests-mock",
   "pytest-datafiles",
   "pytest-xdist",

upgini-1.2.86.dev1/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.2.86.dev1"

{upgini-1.2.86a1 → upgini-1.2.86.dev1}/src/upgini/autofe/feature.py

@@ -8,6 +8,7 @@ from pandas._typing import DtypeObj
 
 from upgini.autofe.all_operators import find_op
 from upgini.autofe.operator import Operator, PandasOperator
+from upgini.autofe.timeseries.base import TimeSeriesBase
 from upgini.autofe.utils import pydantic_dump_method, pydantic_parse_method
 
 
@@ -162,7 +163,11 @@ class Feature:
         if self.cached_display_name is not None and cache:
             return self.cached_display_name
 
-        should_stack_op = not isinstance(self.children[-1], Column) if self.op.is_unary or self.op.is_vector else False
+        should_stack_op = (
+            not isinstance(self.children[-1], Column)
+            if self.op.is_unary or isinstance(self.op, TimeSeriesBase)
+            else False
+        )
         components = []
 
         if self.alias:

{upgini-1.2.86a1 → upgini-1.2.86.dev1}/src/upgini/utils/datetime_utils.py

@@ -121,31 +121,31 @@ class DateTimeSearchKeyConverter:
                 df[cos_feature] = np.cos(2 * np.pi * df[column] / period)
                 self.generated_features.append(cos_feature)
 
-        df["quarter"] = df[self.date_column].dt.quarter
+        # df["quarter"] = df[self.date_column].dt.quarter
 
-        # Calculate the start date of the quarter for each timestamp
-        df["quarter_start"] = df[self.date_column].dt.to_period("Q").dt.start_time
+        # # Calculate the start date of the quarter for each timestamp
+        # df["quarter_start"] = df[self.date_column].dt.to_period("Q").dt.start_time
 
-        # Calculate the day in the quarter
-        df["day_in_quarter"] = (df[self.date_column] - df["quarter_start"]).dt.days + 1
+        # # Calculate the day in the quarter
+        # df["day_in_quarter"] = (df[self.date_column] - df["quarter_start"]).dt.days + 1
 
-        # Vectorized calculation of days_in_quarter
-        quarter = df["quarter"]
-        start = df["quarter_start"]
-        year = start.dt.year
-        month = start.dt.month
+        # # Vectorized calculation of days_in_quarter
+        # quarter = df["quarter"]
+        # start = df["quarter_start"]
+        # year = start.dt.year
+        # month = start.dt.month
 
-        quarter_end_year = np.where(quarter == 4, year + 1, year)
-        quarter_end_month = np.where(quarter == 4, 1, month + 3)
+        # quarter_end_year = np.where(quarter == 4, year + 1, year)
+        # quarter_end_month = np.where(quarter == 4, 1, month + 3)
 
-        end = pd.to_datetime({"year": quarter_end_year, "month": quarter_end_month, "day": 1})
-        end.index = df.index
+        # end = pd.to_datetime({"year": quarter_end_year, "month": quarter_end_month, "day": 1})
+        # end.index = df.index
 
-        df["days_in_quarter"] = (end - start).dt.days
+        # df["days_in_quarter"] = (end - start).dt.days
 
-        add_cyclical_features(df, "day_in_quarter", df["days_in_quarter"])  # Days in the quarter
+        # add_cyclical_features(df, "day_in_quarter", df["days_in_quarter"])  # Days in the quarter
 
-        df.drop(columns=["quarter", "quarter_start", "day_in_quarter", "days_in_quarter"], inplace=True)
+        # df.drop(columns=["quarter", "quarter_start", "day_in_quarter", "days_in_quarter"], inplace=True)
 
         df[seconds] = (df[self.date_column] - df[self.date_column].dt.floor("D")).dt.seconds
 

{upgini-1.2.86a1 → upgini-1.2.86.dev1}/src/upgini/utils/email_utils.py

@@ -36,11 +36,11 @@ class EmailDomainGenerator:
         self.generated_features = []
 
     def generate(self, df: pd.DataFrame) -> pd.DataFrame:
-        for email_col in self.email_columns:
-            domain_feature = email_col + self.DOMAIN_SUFFIX
-            if domain_feature not in df.columns:
-                df[domain_feature] = df[email_col].apply(self._email_to_domain).astype("string")
-                self.generated_features.append(domain_feature)
+        # for email_col in self.email_columns:
+        #     domain_feature = email_col + self.DOMAIN_SUFFIX
+        #     if domain_feature not in df.columns:
+        #         df[domain_feature] = df[email_col].apply(self._email_to_domain).astype("string")
+        #         self.generated_features.append(domain_feature)
         return df
 
     @staticmethod

upgini-1.2.86a1/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.2.86a1"