Unverified Commit 59baae99 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

remove debug artifacts from prototype dataset tests (#4958)

parent 29e0f66a
......@@ -144,9 +144,6 @@ class MNISTFakedata:
@classmethod
def _create_binary_file(cls, root, filename, *, num_samples, shape, dtype, compressor, low=0, high):
with compressor(root / filename, "wb") as fh:
if dtype != torch.uint8:
print()
for meta in (cls._magic(dtype, len(shape)), num_samples, *shape):
fh.write(cls._encode(meta))
......
......@@ -34,8 +34,6 @@ def dataset_parametrization(*names, decoder=to_bytes):
params = []
for name in names:
for config in datasets.info(name)._configs:
if name == "imagenet" and config.split == "test":
print()
id = f"{name}-{'-'.join([str(value) for value in config.values()])}"
dataset, mock_info = builtin_dataset_mocks.load(name, decoder=decoder, **config)
params.append(pytest.param(dataset, mock_info, id=id))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment