Browse Source

Remove iterable

ogert 3 years ago
parent
commit
a8f4a765eb
1 changed files with 3 additions and 2 deletions
  1. 3 2
      cdplib/pipeline_selector/PipelineSelector.py

+ 3 - 2
cdplib/pipeline_selector/PipelineSelector.py

@@ -34,7 +34,7 @@ else:
     # from typing_extensions import *
     print("I have python version {}.{} and will import typing_extensions".format(sys.version_info.major, sys.version_info.minor))
     from typing_extensions import Callable, TypedDict,\
-    Literal, Iterable, Tuple, Union
+    Literal, Tuple, Union
 
 import functools
 from sklearn.pipeline import Pipeline
@@ -336,7 +336,8 @@ class PipelineSelector(ABC):
                     = None,
                     y_val: Union[pd.DataFrame, pd.Series, np.ndarray]
                     = None,
-                    cv: Union[Iterable[Tuple[List[int], List[int]]]]
+                    # cv: Union[Iterable[Tuple[List[int], List[int]]]]
+                    cv
                     = None) -> None:
         '''
         :param array X_train: data on which