fix numerical encoder sign none handling

mindsdb · Jun 13, 2023 · 8c7b78f · 8c7b78f
1 parent 3babcd9
commit 8c7b78f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lightwood/encoder/numeric/numeric.py b/lightwood/encoder/numeric/numeric.py
@@ -58,7 +58,7 @@ def encode(self, data: Union[np.ndarray, pd.Series]):
             data = data.values
 
         if not self.positive_domain:
-            sign_data = np.nan_to_num(data, nan=0, posinf=0, neginf=0)
+            sign_data = np.nan_to_num(data.astype(float), nan=0, posinf=0, neginf=0)
             sign = np.vectorize(self._sign_fn, otypes=[float])(sign_data)
         else:
             sign = np.zeros(len(data))