Skip to content

Commit 733328d

Browse files
authored
fix test after WorkerInfo update in pytorch/pytorch (#1528)
1 parent 422e2fb commit 733328d

File tree

1 file changed

+0
-9
lines changed

1 file changed

+0
-9
lines changed

test/stateful_dataloader/test_dataloader.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -959,15 +959,6 @@ def _test_worker_info_init_fn(worker_id):
959959
dataset = worker_info.dataset
960960
assert isinstance(dataset, TestWorkerInfoDataset), "worker_info should have correct dataset copy"
961961
assert not hasattr(dataset, "value"), "worker_info should have correct dataset copy"
962-
# test that WorkerInfo attributes are read-only
963-
try:
964-
worker_info.id = 3999
965-
except RuntimeError as e:
966-
assert str(e) == "Cannot assign attributes to WorkerInfo objects"
967-
try:
968-
worker_info.a = 3
969-
except RuntimeError as e:
970-
assert str(e) == "Cannot assign attributes to WorkerInfo objects"
971962
for k in ["id", "num_workers", "seed", "dataset"]:
972963
assert f"{k}=" in repr(worker_info)
973964
dataset.value = [worker_id, os.getpid()]

0 commit comments

Comments
 (0)