diff --git a/autoPyTorch/datasets/resampling_strategy.py b/autoPyTorch/datasets/resampling_strategy.py index 860adadaa..1c8fea5fd 100644 --- a/autoPyTorch/datasets/resampling_strategy.py +++ b/autoPyTorch/datasets/resampling_strategy.py @@ -64,6 +64,7 @@ def is_stratified(self) -> bool: class HoldoutValTypes(IntEnum): + """TODO: change to enum using functools.partial""" """The type of hold out validation (refer to CrossValTypes' doc-string)""" holdout_validation = 6 stratified_holdout_validation = 7