소스 검색

fixed np.ndarray type definition

tanja 3 년 전
부모
커밋
46a0090ea8
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      cdplib/pipeline_selector/PipelineSelector.py

+ 2 - 2
cdplib/pipeline_selector/PipelineSelector.py

@@ -264,8 +264,8 @@ class PipelineSelector(ABC):
 
         else:
             try:
-                assert(isinstance(X_train, (pd.DataFrame, np.array)) and
-                       isinstance(y_train, (pd.Series, np.array,
+                assert(isinstance(X_train, (pd.DataFrame, np.ndarray)) and
+                       isinstance(y_train, (pd.Series, np.ndarray,
                                             pd.DataFrame, NoneType)) and
                        (X_val is None) and (y_val is None))
             except AssertionError: