diff --git a/tests/e2e/mnist.py b/tests/e2e/mnist.py index 143a6b6c9..ba5ec96fb 100644 --- a/tests/e2e/mnist.py +++ b/tests/e2e/mnist.py @@ -15,6 +15,7 @@ import os import torch +import torchvision.datasets.utils as utils import requests from pytorch_lightning import LightningModule, Trainer from pytorch_lightning.callbacks.progress import TQDMProgressBar @@ -138,6 +139,11 @@ def configure_optimizers(self): #################### def prepare_data(self): + def check_md5(fpath, md5=None): + return True + + utils.check_md5 = check_md5 + # download print("Downloading MNIST dataset...") diff --git a/tests/e2e/mnist_pip_requirements.txt b/tests/e2e/mnist_pip_requirements.txt index 907e9a52d..05ed58168 100644 --- a/tests/e2e/mnist_pip_requirements.txt +++ b/tests/e2e/mnist_pip_requirements.txt @@ -1,4 +1,4 @@ pytorch_lightning==1.9.5 torchmetrics==0.9.1 -torchvision==0.12.0 +torchvision==0.19.0 minio