Fix pyright re-export issues

Author: Avasam

.github/workflows/test.yml

@@ -37,9 +37,10 @@ jobs:
         uses: jakebailey/pyright-action@v2
         with:
           pylance-version: latest-prerelease
+          python-version: ${{ matrix.python-version }}
 
       - name: Run mypy tests
-        run: python -m mypy .
+        run: python -m mypy . --python-version=${{ matrix.python-version }}
 
       - name: Run stubtest
         run: python tests/run_stubtest.py

pyproject.toml

@@ -117,7 +117,7 @@ extra-standard-library = [
 ]
 
 [tool.pyright]
-exclude = ["build", ".git", ".venv"]
+exclude = ["build", ".git", ".venv*"]
 stubPath = "./stubs"
 pythonversion = "3.9" # Target oldest supported Python version
 typeCheckingMode = "standard"

stubs/sklearn/ensemble/_hist_gradient_boosting/_bitset.pyi

@@ -1,6 +1,4 @@
 import numpy as np
 
-from .common import X_BINNED_DTYPE_C
-
-def set_bitset_memoryview(bitset: np.ndarray, val: X_BINNED_DTYPE_C) -> None: ...
+def set_bitset_memoryview(bitset: np.ndarray, val: np.uint8) -> None: ...
 def set_raw_bitset_from_binned_bitset(raw_bitset: np.ndarray, binned_bitset: np.ndarray, categories: np.ndarray) -> None: ...

stubs/sklearn/ensemble/_hist_gradient_boosting/binning.pyi

@@ -1,24 +1,11 @@
 from collections.abc import Sequence
 from typing_extensions import Self
 
-import numpy as np
 from numpy import ndarray, uint8
 from numpy.random import RandomState
 
 from ..._typing import ArrayLike, Int, MatrixLike
 from ...base import BaseEstimator, TransformerMixin
-from ...utils import check_array as check_array, check_random_state as check_random_state
-from ...utils.fixes import percentile as percentile
-from ...utils.validation import check_is_fitted as check_is_fitted
-from ._bitset import set_bitset_memoryview as set_bitset_memoryview
-from .common import (
-    ALMOST_INF as ALMOST_INF,
-    X_BINNED_DTYPE as X_BINNED_DTYPE,
-    X_BITSET_INNER_DTYPE as X_BITSET_INNER_DTYPE,
-    X_DTYPE as X_DTYPE,
-)
-
-# Author: Nicolas Hug
 
 class _BinMapper(TransformerMixin, BaseEstimator):
     missing_values_bin_idx_: uint8 = ...

stubs/sklearn/ensemble/_hist_gradient_boosting/gradient_boosting.pyi

@@ -1,39 +1,13 @@
-import itertools
-import warnings
 from abc import ABC, abstractmethod
 from collections.abc import Mapping, Sequence
-from functools import partial as partial
-from numbers import Integral as Integral, Real as Real
-from timeit import default_timer as time
 from typing import Callable, ClassVar, Literal
 from typing_extensions import Self
 
-import numpy as np
 from numpy import ndarray
 from numpy.random import RandomState
 
-from ..._loss.loss import (
-    BaseLoss as BaseLoss,
-    HalfBinomialLoss as HalfBinomialLoss,
-    HalfMultinomialLoss as HalfMultinomialLoss,
-    HalfPoissonLoss as HalfPoissonLoss,
-    PinballLoss as PinballLoss,
-)
 from ..._typing import ArrayLike, Float, Int, MatrixLike
-from ...base import BaseEstimator, ClassifierMixin, RegressorMixin, is_classifier as is_classifier
-from ...metrics import check_scoring as check_scoring
-from ...model_selection import train_test_split as train_test_split
-from ...preprocessing import LabelEncoder as LabelEncoder
-from ...utils import (
-    check_random_state as check_random_state,
-    compute_sample_weight as compute_sample_weight,
-    resample as resample,
-)
-from ...utils._param_validation import Interval as Interval, StrOptions as StrOptions
-from ...utils.multiclass import check_classification_targets as check_classification_targets
-from ...utils.validation import check_consistent_length as check_consistent_length, check_is_fitted as check_is_fitted
-from .common import G_H_DTYPE as G_H_DTYPE, X_DTYPE as X_DTYPE, Y_DTYPE as Y_DTYPE
-from .grower import TreeGrower as TreeGrower
+from ...base import BaseEstimator, ClassifierMixin, RegressorMixin
 
 class BaseHistGradientBoosting(BaseEstimator, ABC):
     _parameter_constraints: ClassVar[dict] = ...

stubs/sklearn/ensemble/_hist_gradient_boosting/grower.pyi

@@ -1,23 +1,11 @@
-import numbers
-from heapq import heappop as heappop, heappush as heappush
-from timeit import default_timer as time
 from typing import ClassVar
 
-import numpy as np
 from numpy import ndarray
 
 from ..._typing import ArrayLike, Float, Int, MatrixLike
-from ._bitset import set_raw_bitset_from_binned_bitset as set_raw_bitset_from_binned_bitset
-from .common import (
-    PREDICTOR_RECORD_DTYPE as PREDICTOR_RECORD_DTYPE,
-    X_BITSET_INNER_DTYPE as X_BITSET_INNER_DTYPE,
-    Y_DTYPE as Y_DTYPE,
-    MonotonicConstraint as MonotonicConstraint,
-)
 from .histogram import HistogramBuilder
 from .predictor import TreePredictor
 from .splitting import SplitInfo, Splitter
-from .utils import sum_parallel as sum_parallel
 
 EPS = ...  # to avoid zero division errors
 

stubs/sklearn/ensemble/_hist_gradient_boosting/predictor.pyi

@@ -1,10 +1,6 @@
-import numpy as np
 from numpy import ndarray
 
 from ..._typing import ArrayLike, Int, MatrixLike
-from .common import Y_DTYPE as Y_DTYPE
-
-# Author: Nicolas Hug
 
 class TreePredictor:
     def __init__(