Merge pull request optuna#4941 from Alnusjaponica/fix-type-annotation

Fix type annotation
HideakiImamura · Sep 29, 2023 · d3fb04f · d3fb04f
2 parents 83f3cbf + ee42e7e
commit d3fb04f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/optuna/integration/sklearn.py b/optuna/integration/sklearn.py
@@ -48,11 +48,11 @@
 if not _imports.is_successful():
     BaseEstimator = object  # NOQA
 
-ArrayLikeType = Union[List, np.ndarray, "pd.Series", spmatrix]
+ArrayLikeType = Union[List, np.ndarray, "pd.Series", "spmatrix"]
 OneDimArrayLikeType = Union[List[float], np.ndarray, "pd.Series"]
-TwoDimArrayLikeType = Union[List[List[float]], np.ndarray, "pd.DataFrame", spmatrix]
-IterableType = Union[List, "pd.DataFrame", np.ndarray, "pd.Series", spmatrix, None]
-IndexableType = Union[Iterable, None]
+TwoDimArrayLikeType = Union[List[List[float]], np.ndarray, "pd.DataFrame", "spmatrix"]
+IterableType = Union[List, "pd.DataFrame", np.ndarray, "pd.Series", "spmatrix", None]
+IndexableType = Optional[Iterable]
 
 _logger = logging.get_logger(__name__)