Skip to content

Commit 3923ad9

Browse files
committed
remove redundant call to load datamanager
1 parent 3918d97 commit 3923ad9

File tree

1 file changed

+1
-1
lines changed
  • autoPyTorch/pipeline/components/training/trainer

1 file changed

+1
-1
lines changed

autoPyTorch/pipeline/components/training/trainer/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -492,7 +492,7 @@ def _get_train_label(self, X: Dict[str, Any]) -> List[int]:
492492
Verifies and validates the labels from train split.
493493
"""
494494
# Ensure that the split is not missing any class.
495-
labels: List[int] = X['y_train'][X['backend'].load_datamanager().splits[X['split_id']][0]]
495+
labels: List[int] = X['y_train'][X['train_indices']]
496496
if STRING_TO_TASK_TYPES[X['dataset_properties']['task_type']] in CLASSIFICATION_TASKS:
497497
unique_labels = len(np.unique(labels))
498498
if unique_labels < X['dataset_properties']['output_shape']:

0 commit comments

Comments
 (0)