upgini 1.2.13__py3-none-any.whl → 1.2.13a2__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.2.13"
+__version__ = "1.2.13a2"

upgini/autofe/binary.py

@@ -132,7 +132,7 @@ class CombineThenFreq(PandasOperand):
         self._loc(temp, value_counts)
 
 
-class Distance(PandasOperand):
+class Distance:
     name: str = "dist"
     is_binary: bool = True
     output_type: Optional[str] = "float"
@@ -140,21 +140,27 @@ class Distance(PandasOperand):
     has_symmetry_importance: bool = True
 
     def calculate_binary(self, left: pd.Series, right: pd.Series) -> pd.Series:
+        # Handle None values by replacing them with 0 in the dot product and norm calculations
+        left = left.apply(lambda x: np.array(x) if x is not None else np.zeros_like(right[0]))
+        right = right.apply(lambda x: np.array(x) if x is not None else np.zeros_like(left[0]))
+
         return pd.Series(
             1 - self.__dot(left, right) / (self.__norm(left) * self.__norm(right)), index=left.index
-        ).astype(np.float64)
+        )
 
     # row-wise dot product, handling None values
     def __dot(self, left: pd.Series, right: pd.Series) -> pd.Series:
-        left = left.apply(lambda x: np.array(x))
-        right = right.apply(lambda x: np.array(x))
-        res = (left.dropna() * right.dropna()).apply(np.sum)
-        res = res.reindex(left.index.union(right.index))
+        left = left.apply(lambda x: np.array(x) if x is not None else np.zeros_like(right[0]))
+        right = right.apply(lambda x: np.array(x) if x is not None else np.zeros_like(left[0]))
+
+        # Perform element-wise multiplication and handle missing values
+        res = (left * right).apply(np.sum)
         return res
 
     # Calculate the norm of a vector, handling None values
     def __norm(self, vector: pd.Series) -> pd.Series:
-        vector = vector.fillna(np.nan)
+        # Replace None with a zero vector
+        vector = vector.apply(lambda x: np.array(x) if x is not None else np.zeros_like(vector[0]))
         return np.sqrt(self.__dot(vector, vector))
 
 

{upgini-1.2.13.dist-info → upgini-1.2.13a2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.13
+Version: 1.2.13a2
 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.13.dist-info → upgini-1.2.13a2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-upgini/__about__.py,sha256=rQSlPcfj4yT4krIq6epTVQyBzIX4etVOgfupVkM-RnU,23
+upgini/__about__.py,sha256=hwBCVZvUNp4Oos-G-FVBndpBVHIsC2I6VH0LltckZ0Y,25
 upgini/__init__.py,sha256=M64LwQTBa-5Jz24Zm2h8rWwlKQQ1J8nP7gGgIciS0WU,589
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=olZ-OHSfBNoBSCo7R5t7uCLukI2nO7afpx_A-HCiJLk,31067
@@ -15,7 +15,7 @@ upgini/ads_management/__init__.py,sha256=qzyisOToVRP-tquAJD1PblZhNtMrOB8FiyF9Jvf
 upgini/ads_management/ads_manager.py,sha256=igVbN2jz80Umb2BUJixmJVj-zx8unoKpecVo-R-nGdw,2648
 upgini/autofe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 upgini/autofe/all_operands.py,sha256=cCCB44qvkmuWyiRM5Xykx8tkHPIjQthrWyj67STWN80,2578
-upgini/autofe/binary.py,sha256=zMhtHVuGUAFLUqem-XiXqJj-GRXxS88tdz8tFuDfSNM,7659
+upgini/autofe/binary.py,sha256=bszs7s7IqymOgaNAcR-4MIIV_DHUka4YZkTX7bsuFr8,8104
 upgini/autofe/date.py,sha256=OpFc3Al0xO3qlESn2Uokfxw51ArVqmh3xngWwdrsaqE,9762
 upgini/autofe/feature.py,sha256=eL7wABUhDKZzv3E-RPJNcyGwSfB0UptcfU2RbvsOks4,15082
 upgini/autofe/groupby.py,sha256=r-xl_keZZgm_tpiEoDhjYSkT6NHv7a4cRQR4wJ4uCp8,3263
@@ -57,7 +57,7 @@ upgini/utils/sklearn_ext.py,sha256=13jQS_k7v0aUtudXV6nGUEWjttPQzAW9AFYL5wgEz9k,4
 upgini/utils/target_utils.py,sha256=BVtDmrmFMKerSUWaNOIEdzsYHIFiODdpnWbE50QDPDc,7864
 upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
 upgini/utils/warning_counter.py,sha256=dIWBB4dI5XRRJZudvIlqlIYKEiwLLPcXarsZuYRt338,227
-upgini-1.2.13.dist-info/METADATA,sha256=IRJWMi0M4nUgCqMwp4kffx8QXgR1DJ2VsqH5Y7-nQ2E,48577
-upgini-1.2.13.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-upgini-1.2.13.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.2.13.dist-info/RECORD,,
+upgini-1.2.13a2.dist-info/METADATA,sha256=tZl3cZEAUiIpzU1tZ5TPM3Q9hRCR54whSZp3z95rXss,48579
+upgini-1.2.13a2.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+upgini-1.2.13a2.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.2.13a2.dist-info/RECORD,,