upgini 1.2.7__tar.gz → 1.2.8__tar.gz

{upgini-1.2.7 → upgini-1.2.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.7
+Version: 1.2.8
 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.8/src/upgini/__about__.py

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

{upgini-1.2.7 → upgini-1.2.8}/src/upgini/features_enricher.py

@@ -2095,7 +2095,9 @@ class FeaturesEnricher(TransformerMixin):
             features_not_to_pass = [
                 c
                 for c in df.columns
-                if c not in search_keys.keys() and c not in features_for_transform and c != ENTITY_SYSTEM_RECORD_ID
+                if c not in search_keys.keys()
+                and c not in features_for_transform
+                and c not in [ENTITY_SYSTEM_RECORD_ID, SEARCH_KEY_UNNEST]
             ]
 
             if add_fit_system_record_id:

{upgini-1.2.7 → upgini-1.2.8}/src/upgini/utils/ip_utils.py

@@ -57,17 +57,17 @@ class IpSearchKeyConverter:
         except ValueError:
             pass
 
-    @staticmethod
-    def _is_ipv4(ip: Optional[_BaseAddress]):
-        return ip is not None and (
-            isinstance(ip, IPv4Address) or (isinstance(ip, IPv6Address) and ip.ipv4_mapped is not None)
-        )
-
-    @staticmethod
-    def _to_ipv4(ip: Optional[_BaseAddress]) -> Optional[IPv4Address]:
-        if isinstance(ip, IPv4Address):
-            return ip
-        return None
+    # @staticmethod
+    # def _is_ipv4(ip: Optional[_BaseAddress]):
+    #     return ip is not None and (
+    #         isinstance(ip, IPv4Address) or (isinstance(ip, IPv6Address) and ip.ipv4_mapped is not None)
+    #     )
+
+    # @staticmethod
+    # def _to_ipv4(ip: Optional[_BaseAddress]) -> Optional[IPv4Address]:
+    #     if isinstance(ip, IPv4Address):
+    #         return ip
+    #     return None
 
     @staticmethod
     def _to_ipv6(ip: Optional[_BaseAddress]) -> Optional[IPv6Address]:
@@ -87,10 +87,10 @@ class IpSearchKeyConverter:
             raise ValidationError(self.bundle.get("invalid_ip").format(self.ip_column))
 
         # legacy support
-        ipv4 = self.ip_column + "_v4"
-        df[ipv4] = df[self.ip_column].apply(self._to_ipv4).apply(self._ip_to_int).astype("Int64")
-        self.search_keys[ipv4] = SearchKey.IP
-        self.columns_renaming[ipv4] = original_ip
+        # ipv4 = self.ip_column + "_v4"
+        # df[ipv4] = df[self.ip_column].apply(self._to_ipv4).apply(self._ip_to_int).astype("Int64")
+        # self.search_keys[ipv4] = SearchKey.IP
+        # self.columns_renaming[ipv4] = original_ip
 
         ipv6 = self.ip_column + "_v6"
         df[ipv6] = (

upgini-1.2.7/src/upgini/__about__.py

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