pybear-dask 0.2.0__tar.gz → 0.2.1__tar.gz

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/CHANGELOG.md

@@ -6,16 +6,34 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
 
-
-## [0.2.0] - 2025-07-31
+## [0.2.1] - 2025-08-16
 
 ### Added
-- base:
+- None
+
+### Changed
+- Optional and Union type hints replaced with pipes
+- numbers.Integral in docs and type hints changed to int
+
+### Deprecated
+- None
+
+### Removed
+- None
 
+### Fixed
+- None
+
+### Security
+- None
+
+## [0.2.0] - 2025-07-28
+
+### Added
+    - base:
         is_classifier
 
-- model_selection:
-    
+    - model_selection: 
         GSTCVDask
         AutoGridSearchCVDask
         AutoGSTCVDask

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pybear-dask
-Version: 0.2.0
+Version: 0.2.1
 Summary: Python modules for data analytics applications with dask
 License: BSD-3-Clause
 Keywords: python data analytics pybear dask
@@ -16,7 +16,6 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: dask (<2025.1)
 Requires-Dist: dask-ml (<2025.1)
 Requires-Dist: distributed (<2025.1)
-Requires-Dist: polars-lts-cpu (>=1.31.0,<2.0.0)
 Requires-Dist: pybear (>=0.2.0)
 Requires-Dist: scikit-learn (<1.7)
 Project-URL: Homepage, https://github.com/PylarBear/pybear-dask
@@ -71,11 +70,17 @@ pybear-dask
    :target: https://pepy.tech/project/pybear-dask/
    :alt: PyPI Downloads
 
+|DOI|
+
+.. |DOI| image:: https://zenodo.org/badge/1009051313.svg
+   :target: https://doi.org/10.5281/zenodo.16548280
+   :alt: DOI
+
 .. |PythonVersion| replace:: >=3.10, <3.13
 .. |DaskVersion| replace:: <2025.1.0
 .. |DaskMLVersion| replace:: <2025.1.0
 .. |DistributedVersion| replace:: <2025.1.0
-.. |PybearVersion| replace:: >=0.1.22
+.. |PybearVersion| replace:: >=0.2.0
 .. |PytestVersion| replace:: >=7.0.0
 
 

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/README.rst

@@ -46,11 +46,17 @@ pybear-dask
    :target: https://pepy.tech/project/pybear-dask/
    :alt: PyPI Downloads
 
+|DOI|
+
+.. |DOI| image:: https://zenodo.org/badge/1009051313.svg
+   :target: https://doi.org/10.5281/zenodo.16548280
+   :alt: DOI
+
 .. |PythonVersion| replace:: >=3.10, <3.13
 .. |DaskVersion| replace:: <2025.1.0
 .. |DaskMLVersion| replace:: <2025.1.0
 .. |DistributedVersion| replace:: <2025.1.0
-.. |PybearVersion| replace:: >=0.1.22
+.. |PybearVersion| replace:: >=0.2.0
 .. |PytestVersion| replace:: >=7.0.0
 
 

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pybear-dask"
-version = "0.2.0"
+version = "0.2.1"
 description = "Python modules for data analytics applications with dask"
 license = "BSD-3-clause"
 authors = ["Bill Sousa <WKSJR99@msn.com>"]
@@ -34,7 +34,6 @@ dask-ml = "<2025.1"
 distributed = "<2025.1"
 pybear = ">=0.2.0"    #{ version = ">=0.1.22", source = "testpypi" }
 scikit-learn = "<1.7"
-polars-lts-cpu = "^1.31.0"
 
 
 [tool.poetry.group.dev.dependencies]

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/_version.py

@@ -5,21 +5,15 @@
 #
 
 
-from typing import Tuple
-from typing_extensions import Union
-import numpy as np
 
+import numpy as np
 
-try:
-    from importlib.metadata import version as get_version
-except ImportError:
-    # For older Python versions, use importlib-metadata package
-    from importlib_metadata import version as get_version
+from importlib.metadata import version as get_version
 
 
 version: str
 __version__: str
-VERSION_TUPLE = Tuple[Union[int, str], ...]
+VERSION_TUPLE = tuple[int | str, ...]
 version_tuple: VERSION_TUPLE
 __version_tuple__: VERSION_TUPLE
 

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/model_selection/GSTCV/_GSTCVDask/GSTCVDask.py

@@ -7,13 +7,9 @@
 
 
 from typing import (
+    Any,
     Callable,
     Iterable,
-    Optional
-)
-from typing_extensions import (
-    Any,
-    Union
 )
 from ._type_aliases import (
     DaskXType,
@@ -111,7 +107,7 @@ class GSTCVDask(_GSTCVMixin):
         is explicitly designed for use with dask objects (estimators,
         arrays, and dataframes.) pybear `GSTCV` is recommended for
         non-dask classifiers.
-    param_grid : Union[ParamGridInputType, ParamGridsInputType]
+    param_grid : ParamGridInputType | ParamGridsInputType
         Required. A dictionary with hyperparameters names (str) as keys
         and list-likes of respective settings to try as values. Can also
         be a list-like of such dictionaries, and the grids spanned by
@@ -210,7 +206,7 @@ class GSTCVDask(_GSTCVMixin):
 
         def your_metric_wrapper(y_true, y_pred):
             return your_metric(y_true, y_pred, **hard_coded_kwargs)
-    iid : Optional[bool], default=True
+    iid : bool, default=True
         `iid` is ignored when `cv` is an iterable. Indicates whether
         the data's examples are believed to have random distribution
         (True) or if the examples are organized non-randomly in some
@@ -219,7 +215,7 @@ class GSTCVDask(_GSTCVMixin):
         this can be an expensive process. Otherwise, if the data is iid,
         KFold can handle the data as contiguous chunks which is much
         more efficient.
-    refit : Optional[Union[bool, str, Callable]], default=True
+    refit : bool | str | Callable, default=True
         After the grid search is done, fit the whole dataset on the
         estimator using the best found hyperparameters and expose this
         fitted estimator via the :attr:`best_estimator_` attribute. Also,
@@ -252,7 +248,7 @@ class GSTCVDask(_GSTCVMixin):
 
         See the `scoring` parameter to know more about multiple metric
         evaluation.
-    cv : Optional[Union[numbers.Integral, Iterable, None]], default=None
+    cv : int | Iterable | None, default=None
         Sets the cross-validation splitting strategy.
 
         Possible inputs for cv are:
@@ -269,21 +265,21 @@ class GSTCVDask(_GSTCVMixin):
         of iterables. `GSTCVDask` will catch out of range indices and
         raise an error but any validation beyond that is up to the user
         outside of `GSTCVDask`.
-    verbose : Optional[numbers.Real], default=0
+    verbose : numbers.Real, default=0
         The amount of verbosity to display to screen during the grid
         search. Accepts integers from 0 to 10. 0 means no information
         displayed to the screen, 10 means full verbosity. Non-numbers
         are rejected. Boolean False is set to 0, boolean True is set to
         10. Negative numbers are rejected. Numbers greater than 10 are
         set to 10. Floats are rounded to integers.
-    error_score : Optional[ErrorScoreType], default='raise'
+    error_score : ErrorScoreType, default='raise'
         Score to assign if the estimator raises an error while fitting
         on a train fold. If set to ‘raise’, the error is raised. If a
         numeric value is given, a warning is raised and the error score
         value is inserted into the subsequent calculations in place of
         the missing value(s). This parameter does not affect the refit
         step, which will always raise the error.
-    return_train_score : Optional[bool]
+    return_train_score : bool
         If False, the `cv_results_` attribute will not include training
         scores. If True, the train data is scored using all the scorers
         at the best respective threshold(s) found for the test data.
@@ -291,7 +287,7 @@ class GSTCVDask(_GSTCVMixin):
         or over-fittedness. However, computing the scores on the training
         set can be computationally expensive and is not required to
         select the hyperparameters that yield the best performance.
-    scheduler : Optional[Union[Client, Scheduler, None]], default=None
+    scheduler : Client | Scheduler | None, default=None
         A passed scheduler supersedes all other external schedulers.
         When a scheduler is explicitly passed, `GSTCVDask` does not
         perform any validation or verification but allows that to be
@@ -313,14 +309,14 @@ class GSTCVDask(_GSTCVMixin):
         should be handled by the user external to the `GSTCVDask` module.
         As much as possible, dask and distributed objects are allowed to
         flow through without any hard-coded input.
-    n_jobs : Optional[Union[int, None]], default=None
+    n_jobs : int | None, default=None
         Active only if no scheduler is available. That is, if a
         scheduler is not passed to `scheduler`, if no global scheduler
         is available, and if there is no scheduler context manager, only
         then does `n_jobs` become effectual. In this case, `GSTCVDask`
         creates a Client instance using local multiprocessing with
         `n_workers=n_jobs`.
-    cache_cv : Optional[bool], default=True
+    cache_cv : bool, default=True
         Indicates if the train/test folds of the data are to be stored
         when first generated, or if the folds are generated from X, y
         and the KFold indices at each point of need. Caching all the
@@ -449,7 +445,6 @@ class GSTCVDask(_GSTCVMixin):
     refit_time_ : float
         Seconds elapsed when refitting the best model on the whole
         dataset. Only exposed when `refit` is not False.
-
     multimetric_ : bool
         Whether several scoring metrics were used. False if one scorer
         was used, otherwise True. Always exposed after fit.
@@ -490,7 +485,7 @@ class GSTCVDask(_GSTCVMixin):
         Sequence[ParamGridInputType]
 
     ThresholdsInputType:
-        Union[None, numbers.Real, Sequence[numbers.Real]]
+        None | numbers.Real | Sequence[numbers.Real]
 
     DaskSlicerType:
         dask.array.core.Array
@@ -512,27 +507,23 @@ class GSTCVDask(_GSTCVMixin):
         Callable[[Iterable, Iterable], numbers.Real]
 
     ScorerInputType:
-        Union[
-            ScorerNameTypes,
-            Sequence[ScorerNameTypes],
-            ScorerCallableType,
-            dict[str, ScorerCallableType]
-        ]
+        ScorerNameTypes | Sequence[ScorerNameTypes]
+        | ScorerCallableType | dict[str, ScorerCallableType]
 
     RefitCallableType:
-        Callable[[CVResultsType], numbers.Integral]
+        Callable[[CVResultsType], int]
 
     RefitType:
-        Union[bool, ScorerNameTypes, RefitCallableType]
+        bool | ScorerNameTypes | RefitCallableType
 
     DaskSchedulerType:
-        Union[distributed.scheduler.Scheduler, distributed.client.Client]
+        distributed.scheduler.Scheduler | distributed.client.Client
 
     DaskXType:
         Iterable
 
     DaskYType:
-        Union[Sequence[numbers.Integral], None]
+        Sequence[int] | None
 
     CVResultsType:
         dict[str, np.ma.masked_array[Any]]
@@ -590,21 +581,19 @@ class GSTCVDask(_GSTCVMixin):
     def __init__(
         self,
         estimator: ClassifierProtocol,
-        param_grid: Union[ParamGridInputType, ParamGridsInputType],
+        param_grid: ParamGridInputType | ParamGridsInputType,
         *,
-        thresholds: ThresholdsInputType=None,
-        scoring: ScorerInputType='accuracy',
-        iid: Optional[bool]=True,
-        refit: Optional[Union[bool, str, Callable]] = True,
-        cv: Optional[Union[numbers.Integral, Iterable, None]]=None,
-        verbose: Optional[numbers.Real]=0,
-        error_score: Optional[ErrorScoreType]='raise',
-        return_train_score: Optional[bool]=False,
-        scheduler: Optional[
-            Union[distributed.Client, distributed.scheduler.Scheduler, None]
-        ]=None,
-        n_jobs: Optional[Union[numbers.Integral, None]]=None,
-        cache_cv: Optional[bool]=True
+        thresholds: ThresholdsInputType = None,
+        scoring: ScorerInputType = 'accuracy',
+        iid:bool = True,
+        refit:bool | str | Callable = True,
+        cv:int | Iterable | None = None,
+        verbose:numbers.Real = 0,
+        error_score: ErrorScoreType = 'raise',
+        return_train_score:bool = False,
+        scheduler:  distributed.Client | distributed.scheduler.Scheduler | None=None,
+        n_jobs:int | None = None,
+        cache_cv:bool = True
     ) -> None:
         """Initialize the `GSTCVDask` instance."""
 
@@ -689,7 +678,7 @@ class GSTCVDask(_GSTCVMixin):
         else:  # _cv is an iterable, _cond_cv should have made list[tuple]
             self._KFOLD = self._cv
 
-        self._CACHE_CV: Union[None, list[DaskSplitType]] = None
+        self._CACHE_CV: None | list[DaskSplitType] = None
         if self.cache_cv:
             self._CACHE_CV = []
             for (train_idxs, test_idxs) in self._KFOLD:

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/model_selection/GSTCV/_GSTCVDask/_fit/_estimator_fit_params_helper.py

@@ -6,7 +6,7 @@
 
 
 
-from typing_extensions import Any
+from typing import Any
 from .._type_aliases import DaskKFoldType
 
 from dask import compute

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/model_selection/GSTCV/_GSTCVDask/_fit/_fold_splitter.py

@@ -6,8 +6,6 @@
 
 
 
-from typing_extensions import Union
-
 from .._type_aliases import (
     DaskSlicerType,
     DaskSplitType,
@@ -23,7 +21,7 @@ import dask.dataframe as ddf
 def _fold_splitter(
     train_idxs: DaskSlicerType,
     test_idxs: DaskSlicerType,
-    *data_objects: Union[DaskXType, DaskYType]
+    *data_objects: DaskXType | DaskYType
 ) -> tuple[DaskSplitType, ...]:
     """Split given data objects into train / test pairs using the given
     train and test indices.
@@ -44,7 +42,7 @@ def _fold_splitter(
     test_idxs : DaskSlicerType
         1D vector of row indices used to slice test sets out of every
         given data object.
-    *data_objects : Union[DaskXType, DaskYType]
+    *data_objects : DaskXType | DaskYType
         The data objects to slice. Need not be of equal size, and need
         not be completely consumed in the train / test splits. However,
         standard indexing rules apply when slicing by `train_idxs` and

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/model_selection/GSTCV/_GSTCVDask/_fit/_get_kfold.py

@@ -8,7 +8,6 @@
 
 from typing import (
     Iterator,
-    Optional
 )
 from .._type_aliases import (
     DaskXType,
@@ -27,7 +26,7 @@ def _get_kfold(
     _n_splits: int,
     _iid: bool,
     _verbose: int,
-    _y: Optional[DaskYType] = None
+    _y: DaskYType | None = None
 ) -> Iterator[DaskKFoldType]:
     """Use dask_ml KFold to get train / test splits when cv is passed as
     an integer.
@@ -64,7 +63,7 @@ def _get_kfold(
         A number from 0 to 10 indicating the amount of information to
         display to screen during the grid search trials. 0 means no
         output, 10 means full output.
-    _y : Optional[DaskYType]
+    _y : DaskYType | None, default = None
         The target the data is being fit against, to be split in the
         same way as the data.
 

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/model_selection/GSTCV/_GSTCVDask/_fit/_parallelized_fit.py

@@ -6,10 +6,9 @@
 
 
 
-from typing import Literal
-from typing_extensions import (
+from typing import (
     Any,
-    Union
+    Literal
 )
 from .._type_aliases import (
     DaskXType,
@@ -30,7 +29,7 @@ def _parallelized_fit(
     _y_train: DaskYType,
     _estimator: ClassifierProtocol,
     _grid: dict[str, Any],
-    _error_score: Union[numbers.Real, Literal['raise']],
+    _error_score: numbers.Real | Literal['raise'],
     **_fit_params
 ) -> tuple[ClassifierProtocol, float, bool]:
     """Estimator fit method designed for dask parallelism.
@@ -56,7 +55,7 @@ def _parallelized_fit(
     _grid : dict[str, Any]
         The hyperparameter values to be used during this fit. One
         permutation of all the grid search permutations.
-    _error_score : Union[numbers.Real, Literal['raise']]
+    _error_score : numbers.Real | Literal['raise']
         If a training fold excepts during fitting, the exception can be
         allowed to raise by passing the 'raise' literal. Otherwise,
         passing a number-like will cause the exception to be handled,

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/model_selection/GSTCV/_GSTCVDask/_fit/_parallelized_scorer.py

@@ -6,7 +6,6 @@
 
 
 
-from typing_extensions import Union
 from pybear.model_selection.GSTCV._type_aliases import (
     ScorerWIPType,
     ClassifierProtocol,
@@ -35,7 +34,7 @@ def _parallelized_scorer(
     _f_idx: int,
     _SCORER_DICT: ScorerWIPType,
     _THRESHOLDS: ThresholdsWIPType,
-    _error_score: Union[numbers.Real, None],
+    _error_score: numbers.Real | None,
     _verbose: int
 ) -> tuple[MaskedHolderType, MaskedHolderType]:
 
@@ -72,7 +71,7 @@ def _parallelized_scorer(
         from the mother before building cv_results. This is the vector
         of thresholds from the mother that also mothered this search
         permutation.
-    _error_score : Union[numbers.Real, Literal['raise']]
+    _error_score : numbers.Real | Literal['raise']
         If the training fold complementing this test fold excepted during
         fitting and `error_score` was set to the 'raise' literal, this
         module cannot be reached. Otherwise, a number or number-like was
@@ -108,7 +107,7 @@ def _parallelized_scorer(
     assert isinstance(_SCORER_DICT, dict)
     assert all(map(callable, _SCORER_DICT.values()))
     assert isinstance(_THRESHOLDS, list)
-    assert isinstance(_verbose, (int, float))
+    assert isinstance(_verbose, numbers.Real)
     # END validation ** * ** * ** * ** * ** * ** * ** * ** * ** * ** * *
 
     if _verbose >= 5:

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/model_selection/GSTCV/_GSTCVDask/_fit/_parallelized_train_scorer.py

@@ -6,7 +6,6 @@
 
 
 
-from typing_extensions import Union
 from pybear.model_selection.GSTCV._type_aliases import (
     ScorerWIPType,
     ClassifierProtocol,
@@ -34,7 +33,7 @@ def _parallelized_train_scorer(
     _f_idx: int,
     _SCORER_DICT: ScorerWIPType,
     _BEST_THRESHOLDS_BY_SCORER: NDArrayHolderType,
-    _error_score: Union[numbers.Real, None],
+    _error_score: numbers.Real | None,
     _verbose: int
 ) -> MaskedHolderType:
 
@@ -69,7 +68,7 @@ def _parallelized_train_scorer(
         score for each scorer is found. This vector has length n_scorers
         and in each position holds a float indicating the threshold
         value that is the best threshold for that scorer.
-    _error_score : Union[numbers.Real, Literal['raise']]
+    _error_score : numbers.Real | Literal['raise']
         If this training fold excepted during fitting and `error_score`
         was set to the 'raise' literal, this module cannot be reached.
         Otherwise, a number or number-like was passed to `error_score`.
@@ -105,7 +104,7 @@ def _parallelized_train_scorer(
     assert isinstance(_SCORER_DICT, dict)
     assert all(map(callable, _SCORER_DICT.values()))
     assert isinstance(_BEST_THRESHOLDS_BY_SCORER, np.ndarray)
-    assert isinstance(_verbose, (int, float))
+    assert isinstance(_verbose, numbers.Real)
     # END validation ** * ** * ** * ** * ** * ** * ** * ** * ** * ** * *
 
     if _verbose >= 5:

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/model_selection/GSTCV/_GSTCVDask/_param_conditioning/_scheduler.py

@@ -6,11 +6,9 @@
 
 
 
-from typing_extensions import Union
 from .._type_aliases import DaskSchedulerType
 
 import contextlib
-import numbers
 
 from distributed import (
     Client,
@@ -20,8 +18,8 @@ from distributed import (
 
 
 def _cond_scheduler(
-    _scheduler: Union[DaskSchedulerType, None],
-    _n_jobs: Union[numbers.Integral, None]
+    _scheduler: DaskSchedulerType | None,
+    _n_jobs: int | None
 ) -> DaskSchedulerType:
     """Set the dask scheduler.
 
@@ -49,7 +47,7 @@ def _cond_scheduler(
 
     Parameters
     ----------
-    _scheduler : Union[SchedulerType, None]
+    _scheduler : SchedulerType | None
         _scheduler to be validated and used for compute
 
     Returns

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/model_selection/GSTCV/_GSTCVDask/_type_aliases.py

@@ -9,14 +9,9 @@
 from typing import (
     ContextManager,
     Iterable,
-    Sequence
+    Sequence,
+    TypeAlias
 )
-from typing_extensions import (
-    TypeAlias,
-    Union
-)
-
-import numbers
 
 import dask
 import distributed
@@ -24,7 +19,7 @@ import distributed
 
 
 DaskXType: TypeAlias = Iterable
-DaskYType: TypeAlias = Union[Sequence[numbers.Integral], None]
+DaskYType: TypeAlias = Sequence[int] | None
 
 DaskSlicerType: TypeAlias = dask.array.core.Array
 
@@ -32,11 +27,12 @@ DaskKFoldType: TypeAlias = tuple[DaskSlicerType, DaskSlicerType]
 
 DaskSplitType: TypeAlias = tuple[DaskXType, DaskYType]
 
-DaskSchedulerType: TypeAlias = Union[
-    distributed.scheduler.Scheduler,
-    distributed.client.Client,
-    ContextManager  # nullcontext
-]
+DaskSchedulerType: TypeAlias = (
+    distributed.scheduler.Scheduler
+    | distributed.client.Client
+    | ContextManager
+ )  # nullcontext
+
 
 
 

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/model_selection/GSTCV/_GSTCVDask/_validation/_y.py

@@ -18,7 +18,7 @@ import dask.dataframe as ddf
 
 
 def _val_y(
-    _y: Iterable[numbers.Integral]  # not DaskYType... see the notes.
+    _y: Iterable[int]  # not DaskYType... see the notes.
 ) -> None:
     """Validate `_y`.
 

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/model_selection/autogridsearch/AutoGSTCVDask.py

@@ -7,17 +7,10 @@
 
 
 from typing import (
-    Optional,
-    Sequence,
-    Tuple
-)
-from typing_extensions import (
     Any,
-    Union
+    Sequence
 )
 
-import numbers
-
 from pybear.model_selection.autogridsearch.autogridsearch_wrapper import autogridsearch_wrapper
 from pybear.model_selection.autogridsearch import autogridsearch_docs
 
@@ -36,17 +29,17 @@ class AutoGSTCVDask(autogridsearch_wrapper(GSTCVDask)):
         estimator,
         params: dict[
             str,
-            Sequence[Tuple[
+            Sequence[tuple[
                 Sequence[Any],
-                Union[numbers.Integral, Sequence[numbers.Integral]],
+                int | Sequence[int],
                 str
             ]]
         ],
         *,
-        total_passes:Optional[numbers.Integral] = 5,
-        total_passes_is_hard:Optional[bool] = False,
-        max_shifts:Optional[Union[None, numbers.Integral]] = None,
-        agscv_verbose:Optional[bool] = False,
+        total_passes:int = 5,
+        total_passes_is_hard:bool = False,
+        max_shifts:int | None = None,
+        agscv_verbose:bool = False,
         **parent_gscv_kwargs
     ):
 

{pybear_dask-0.2.0 → pybear_dask-0.2.1}/src/pybear_dask/model_selection/autogridsearch/AutoGridSearchCVDask.py

@@ -7,17 +7,10 @@
 
 
 from typing import (
-    Optional,
-    Sequence,
-    Tuple
-)
-from typing_extensions import (
     Any,
-    Union
+    Sequence
 )
 
-import numbers
-
 from pybear.model_selection import autogridsearch_wrapper
 from pybear.model_selection.autogridsearch import autogridsearch_docs
 
@@ -34,19 +27,12 @@ class AutoGridSearchCVDask(autogridsearch_wrapper(dask_GridSearchCV)):
     def __init__(
         self,
         estimator,
-        params: dict[
-            str,
-            Sequence[Tuple[
-                Sequence[Any],
-                Union[numbers.Integral, Sequence[numbers.Integral]],
-                str
-            ]]
-        ],
+        params: dict[str, Sequence[tuple[Sequence[Any], int | Sequence[int], str]]],
         *,
-        total_passes:Optional[numbers.Integral] = 5,
-        total_passes_is_hard:Optional[bool] = False,
-        max_shifts:Optional[Union[None, numbers.Integral]] = None,
-        agscv_verbose:Optional[bool] = False,
+        total_passes:int = 5,
+        total_passes_is_hard:bool = False,
+        max_shifts:int | None = None,
+        agscv_verbose:bool = False,
         **parent_gscv_kwargs
     ):