|
4 | 4 | import itertools |
5 | 5 | import math |
6 | 6 | import sys |
| 7 | +from collections.abc import Sequence |
7 | 8 | from copy import copy, deepcopy |
8 | | -from typing import TYPE_CHECKING, Any, Callable, List, Optional, Sequence, Tuple, Union |
| 9 | +from typing import TYPE_CHECKING, Any, Callable, Optional, Union |
9 | 10 |
|
10 | 11 | import cloudpickle |
11 | 12 | import numpy as np |
|
41 | 42 | # -- types -- |
42 | 43 |
|
43 | 44 | # Commonly used types |
44 | | - Interval: TypeAlias = Union[Tuple[float, float], Tuple[float, float, int]] |
45 | | - NeighborsType: TypeAlias = SortedDict[float, List[Optional[float]]] |
| 45 | + Interval: TypeAlias = Union[tuple[float, float], tuple[float, float, int]] |
| 46 | + NeighborsType: TypeAlias = SortedDict[float, list[Optional[float]]] |
46 | 47 |
|
47 | 48 | # Types for loss_per_interval functions |
48 | | - XsType0: TypeAlias = Tuple[float, float] |
49 | | - YsType0: TypeAlias = Union[Tuple[float, float], Tuple[np.ndarray, np.ndarray]] |
50 | | - XsType1: TypeAlias = Tuple[ |
| 49 | + XsType0: TypeAlias = tuple[float, float] |
| 50 | + YsType0: TypeAlias = Union[tuple[float, float], tuple[np.ndarray, np.ndarray]] |
| 51 | + XsType1: TypeAlias = tuple[ |
51 | 52 | Optional[float], Optional[float], Optional[float], Optional[float] |
52 | 53 | ] |
53 | 54 | YsType1: TypeAlias = Union[ |
54 | | - Tuple[Optional[float], Optional[float], Optional[float], Optional[float]], |
55 | | - Tuple[ |
| 55 | + tuple[Optional[float], Optional[float], Optional[float], Optional[float]], |
| 56 | + tuple[ |
56 | 57 | Optional[np.ndarray], |
57 | 58 | Optional[np.ndarray], |
58 | 59 | Optional[np.ndarray], |
59 | 60 | Optional[np.ndarray], |
60 | 61 | ], |
61 | 62 | ] |
62 | | - XsTypeN: TypeAlias = Tuple[Optional[float], ...] |
| 63 | + XsTypeN: TypeAlias = tuple[Optional[float], ...] |
63 | 64 | YsTypeN: TypeAlias = Union[ |
64 | | - Tuple[Optional[float], ...], Tuple[Optional[np.ndarray], ...] |
| 65 | + tuple[Optional[float], ...], tuple[Optional[np.ndarray], ...] |
65 | 66 | ] |
66 | 67 |
|
67 | 68 |
|
|
0 commit comments