fix bug

stemflow/utils/validation.py

@@ -123,7 +123,7 @@ def check_X_train(X_train):
     if not isinstance(X_train, pd.core.frame.DataFrame):
         raise TypeError(f"Input X should be type 'pd.core.frame.DataFrame'. Got {str(type_X_train)}")
 
-    if np.sum(np.isnan(X_train)) > 0:
+    if np.sum(np.isnan(np.array(X_train))) > 0:
         raise ValueError(
             "NAs (missing values) detected in input data. stemflow do not support NAs input. Consider filling them with values (e.g., -1 or mean values) or removing the rows."
         )
@@ -136,7 +136,7 @@ def check_y_train(y_train):
             f"Input y_train should be type 'pd.core.frame.DataFrame' or 'pd.core.frame.Series', or 'np.ndarray'. Got {str(type_y_train)}"
         )
 
-    if np.sum(np.isnan(y_train)) > 0:
+    if np.sum(np.isnan(np.array(y_train))) > 0:
         raise ValueError("NAs (missing values) detected in input y data. Consider deleting these rows.")