Fix type annotation

HideakiImamura · Sep 24, 2023 · 1b89d98 · 1b89d98
1 parent 0ff4164
commit 1b89d98
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/optuna/integration/sklearn.py b/optuna/integration/sklearn.py
@@ -48,10 +48,10 @@
 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]
+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]
 
 _logger = logging.get_logger(__name__)