Fix mnist_train_test.py issues on Windows (verified locally).

iree-org · Jul 12, 2023 · 6581733 · 6581733
1 parent 523d323
commit 6581733
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/e2e/models/mnist_train_test/mnist_train_test.py b/tests/e2e/models/mnist_train_test/mnist_train_test.py
@@ -43,7 +43,7 @@ def cleanup():
             os.unlink(mmap_vmfb.name)
 
         return load_vm_flatbuffer_file(
-            vmfb_file, driver=args.driver, destroy_callback=cleanup
+            mmap_vmfb.name, driver=args.driver, destroy_callback=cleanup
         )