fix format

aws · Sep 30, 2024 · f1bea3e · f1bea3e
1 parent b53c59b
commit f1bea3e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integ/sagemaker/workflow/test_retry.py b/tests/integ/sagemaker/workflow/test_retry.py
@@ -149,7 +149,7 @@ def test_model_registration_with_model_repack(
     pipeline_name,
     region_name,
     pytorch_training_latest_version,
-    pytorch_training_latest_py_version
+    pytorch_training_latest_py_version,
 ):
     base_dir = os.path.join(DATA_DIR, "pytorch_mnist")
     entry_point = os.path.join(base_dir, "mnist.py")