Adjust CodeFlare SDK tests to run on FIPS

tests/e2e/mnist.py

@@ -12,6 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# Import the MD5 patch utility
+from utils_md5 import apply_md5_patch
+
+# Apply the patch to ignore MD5 check
+apply_md5_patch()
+
 import os
 
 import torch

tests/e2e/utils_md5.py

@@ -0,0 +1,9 @@
+import torchvision.datasets.utils as utils
+
+# Modifying method to ignore MD5 check
+def check_md5(fpath, md5=None):
+    return True
+
+# Apply the modified check_md5 function
+def apply_md5_patch():
+    utils.check_md5 = check_md5