Remove unused mock code for MD5 checks

project-codeflare · Sep 23, 2024 · c6ed1a1 · c6ed1a1
1 parent 868386e
commit c6ed1a1
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/tests/e2e/mnist.py b/tests/e2e/mnist.py
@@ -15,7 +15,6 @@
 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
@@ -139,11 +138,6 @@ 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...")