diff --git a/src/skimage-stubs/filters.pyi b/src/skimage-stubs/filters.pyi index 31a5e8bff056cfe3ef5e18d1188606ec5bb4ed3e..92909ae3c546c584baefa95469d346c1024050b3 100644 --- a/src/skimage-stubs/filters.pyi +++ b/src/skimage-stubs/filters.pyi @@ -3,7 +3,7 @@ Typing stubs for skimage.filters. """ from typing import Optional, Union, Any -from scientific_plots.types_ import Matrix +from scientific_plots.types_ import Matrix, Vector def butterworth( diff --git a/src/sklearn-stubs/ensemble.pyi b/src/sklearn-stubs/ensemble.pyi index 3df757c844b5f668be76a2e41228cf8128f64ba1..55a9b209cf2075ee7a0c5169cbc243627f67154c 100644 --- a/src/sklearn-stubs/ensemble.pyi +++ b/src/sklearn-stubs/ensemble.pyi @@ -4,10 +4,9 @@ Typing stubs for sklearn.ensemble. """ from __future__ import annotations -from typing import Optional, Union, List, Tuple, Dict, TypeVar +from typing import Optional, Union, List, Dict, TypeVar from scientific_plots.types_ import Matrix, Vector as Vector_ -from vector_type import Vector # Assuming Vector is defined in vector_type module Vector = TypeVar("Vector", bound=Union[Vector_, Matrix]) @@ -21,7 +20,7 @@ class IsolationForest: bootstrap: bool = False, n_jobs: Optional[int] = None, behaviour: str = 'deprecated', - random_state: Optional[Union[int, np.random.RandomState]] = None, + random_state: Optional[int] = None, verbose: int = 0, warm_start: bool = False) -> None: ... @@ -45,18 +44,18 @@ class IsolationForest: deep: bool = True) -> Dict[str, Union[int, float, str, bool, None]]: ... - def set_params(self, **params) -> IsolationForest: ... + def set_params(self, params: dict[str, float]) -> IsolationForest: ... # Additional methods for compatibility with BaseEstimator and OutlierMixin - def __getstate__(self) -> dict: ... - def __setstate__(self, state: dict) -> None: ... + def __getstate__(self) -> dict[str, float]: ... + def __setstate__(self, state: dict[str, float]) -> None: ... # Properties @property - def estimators_(self) -> List: ... + def estimators_(self) -> List[float]: ... @property - def estimators_samples_(self) -> List: ... + def estimators_samples_(self) -> List[float]: ... @property def max_samples_(self) -> int: ...