Skip to content

Commit c7fd2d5

Browse files
committed
[fix]: back to the renamed version of CROSS_VAL_FN from temporal SplitFunc typing.
1 parent 77fe0a3 commit c7fd2d5

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

autoPyTorch/datasets/base_dataset.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from abc import ABCMeta
2-
from typing import Any, Dict, List, Optional, Sequence, Tuple, Union, cast, Callable
2+
from typing import Any, Dict, List, Optional, Sequence, Tuple, Union, cast
33

44
import numpy as np
55

@@ -17,12 +17,11 @@
1717
CrossValFunc,
1818
DEFAULT_RESAMPLING_PARAMETERS,
1919
HoldoutValTypes,
20-
HoldOutValFunc
20+
HoldoutValFunc
2121
)
2222
from autoPyTorch.utils.common import FitRequirement, hash_array_or_matrix
2323

2424
BaseDatasetType = Union[Tuple[np.ndarray, np.ndarray], Dataset]
25-
SplitFunc = Callable[[Union[int, float], np.ndarray, Any], List[Tuple[np.ndarray, np.ndarray]]]
2625

2726

2827
def check_valid_data(data: Any) -> None:
@@ -111,7 +110,7 @@ def __init__(
111110
type_check(train_tensors, val_tensors)
112111
self.train_tensors, self.val_tensors, self.test_tensors = train_tensors, val_tensors, test_tensors
113112
self.cross_validators: Dict[str, CrossValFunc] = {}
114-
self.holdout_validators: Dict[str, HoldOutValFunc] = {}
113+
self.holdout_validators: Dict[str, HoldoutValFunc] = {}
115114
self.rng = np.random.RandomState(seed=seed)
116115
self.shuffle = shuffle
117116
self.resampling_strategy = resampling_strategy

0 commit comments

Comments
 (0)