Skip to content

Commit 8398188

Browse files
committed
fixed the for loop in type_check based on samuel's review
1 parent eb95578 commit 8398188

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

autoPyTorch/datasets/base_dataset.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,11 +34,12 @@ def check_valid_data(data: Any) -> None:
3434

3535

3636
def type_check(train_tensors: BaseDatasetType, val_tensors: Optional[BaseDatasetType] = None) -> None:
37-
for train_tensor in train_tensors:
38-
check_valid_data(train_tensor)
37+
"""To avoid unexpected behavior, we use loops over indices."""
38+
for i in range(len(train_tensors)):
39+
check_valid_data(train_tensors[i])
3940
if val_tensors is not None:
40-
for val_tensor in val_tensors:
41-
check_valid_data(val_tensor)
41+
for i in range(len(val_tensors)):
42+
check_valid_data(val_tensors[i])
4243

4344

4445
class TransformSubset(Subset):

0 commit comments

Comments
 (0)