reverted test changes

Signed-off-by: Elias Joseph <[email protected]>

tests/e2e/stablehlo_models/mnist_train_test/mnist_train_test.py

@@ -147,4 +147,4 @@ def test_mnist_training(self):
 
 if __name__ == "__main__":
     args, remaining_args = parse_args()
-    unittest.main(argv=[sys.argv[0]] + remaining_args)
+    unittest.main(argv=[sys.argv[0]] + remaining_args)