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: ...@@ -144,9 +144,6 @@ class MNISTFakedata:
@classmethod @classmethod
def _create_binary_file(cls, root, filename, *, num_samples, shape, dtype, compressor, low=0, high): def _create_binary_file(cls, root, filename, *, num_samples, shape, dtype, compressor, low=0, high):
with compressor(root / filename, "wb") as fh: with compressor(root / filename, "wb") as fh:
if dtype != torch.uint8:
print()
for meta in (cls._magic(dtype, len(shape)), num_samples, *shape): for meta in (cls._magic(dtype, len(shape)), num_samples, *shape):
fh.write(cls._encode(meta)) fh.write(cls._encode(meta))
......
...@@ -34,8 +34,6 @@ def dataset_parametrization(*names, decoder=to_bytes): ...@@ -34,8 +34,6 @@ def dataset_parametrization(*names, decoder=to_bytes):
params = [] params = []
for name in names: for name in names:
for config in datasets.info(name)._configs: 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()])}" id = f"{name}-{'-'.join([str(value) for value in config.values()])}"
dataset, mock_info = builtin_dataset_mocks.load(name, decoder=decoder, **config) dataset, mock_info = builtin_dataset_mocks.load(name, decoder=decoder, **config)
params.append(pytest.param(dataset, mock_info, id=id)) 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