Skip to content

Commit 59baae9

Browse files
authored
remove debug artifacts from prototype dataset tests (#4958)
1 parent 29e0f66 commit 59baae9

File tree

2 files changed

+0
-5
lines changed

2 files changed

+0
-5
lines changed

test/builtin_dataset_mocks.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -144,9 +144,6 @@ def _big_endian_dtype(dtype):
144144
@classmethod
145145
def _create_binary_file(cls, root, filename, *, num_samples, shape, dtype, compressor, low=0, high):
146146
with compressor(root / filename, "wb") as fh:
147-
if dtype != torch.uint8:
148-
print()
149-
150147
for meta in (cls._magic(dtype, len(shape)), num_samples, *shape):
151148
fh.write(cls._encode(meta))
152149

test/test_prototype_builtin_datasets.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,6 @@ def dataset_parametrization(*names, decoder=to_bytes):
3434
params = []
3535
for name in names:
3636
for config in datasets.info(name)._configs:
37-
if name == "imagenet" and config.split == "test":
38-
print()
3937
id = f"{name}-{'-'.join([str(value) for value in config.values()])}"
4038
dataset, mock_info = builtin_dataset_mocks.load(name, decoder=decoder, **config)
4139
params.append(pytest.param(dataset, mock_info, id=id))

0 commit comments

Comments
 (0)