upgini 1.2.13a1__tar.gz → 1.2.13a3__tar.gz

{upgini-1.2.13a1 → upgini-1.2.13a3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.13a1
+Version: 1.2.13a3
 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.13a3/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.2.13a3"

{upgini-1.2.13a1 → upgini-1.2.13a3}/src/upgini/autofe/binary.py

@@ -140,19 +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
         )
 
-    # row-wise dot product
+    # 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:
+        # 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.13a1/src/upgini/__about__.py

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