README.md
pyproject.toml
dataset_with_logits/__init__.py
dataset_with_logits/prediction_dataset.py
dataset_with_logits.egg-info/PKG-INFO
dataset_with_logits.egg-info/SOURCES.txt
dataset_with_logits.egg-info/dependency_links.txt
dataset_with_logits.egg-info/requires.txt
dataset_with_logits.egg-info/top_level.txt
dataset_with_logits/datasets/__init__.py
dataset_with_logits/datasets/base.py
dataset_with_logits/datasets/imagenet.py
dataset_with_logits/datasets/simple.py
dataset_with_logits/utils/__init__.py
dataset_with_logits/utils/download.py
dataset_with_logits/utils/validation.py
tests/test_dataset.py
tests/test_utils.py