From abb982ddf3fc1d8ef0f71d5f13f9f107a873d619 Mon Sep 17 00:00:00 2001
From: oksanabaza <obazylie@redhat.com>
Date: Tue, 10 Sep 2024 13:26:29 +0100
Subject: [PATCH] Adjust CodeFlare SDK tests to run on FIPS

---
 tests/e2e/mnist.py                   | 6 ++++++
 tests/e2e/mnist_pip_requirements.txt | 2 +-
 2 files changed, 7 insertions(+), 1 deletion(-)

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