|
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