fix: fixing format errors

aws · Oct 25, 2024 · 0d7dadf · 0d7dadf
1 parent dc99b48
commit 0d7dadf
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/integ/test_pytorch.py b/tests/integ/test_pytorch.py
@@ -94,6 +94,7 @@ def fixture_training_job_with_latest_inference_version(
         pytorch.fit({"training": _upload_training_data(pytorch)})
         return pytorch.latest_training_job.name
 
+
 @pytest.mark.skip(
     reason="The test is temporarily disabled because it's causing errors with 2.4.0 pytorch version. \
 Please run that manually before the proper fix."
@@ -126,6 +127,7 @@ def test_framework_processing_job_with_deps(
             wait=True,
         )
 
+
 @pytest.mark.skip(
     reason="The test is temporarily disabled because it's causing errors with 2.4.0 pytorch version. \
 Please run that manually before the proper fix."
@@ -149,6 +151,7 @@ def test_fit_deploy(
 
         assert output.shape == (batch_size, 10)
 
+
 @pytest.mark.skip(
     reason="The test is temporarily disabled because it's causing errors with 2.4.0 pytorch version. \
 Please run that manually before the proper fix."
@@ -179,6 +182,7 @@ def test_local_fit_deploy(
     finally:
         predictor.delete_endpoint()
 
+
 @pytest.mark.skip(
     reason="The test is temporarily disabled because it's causing errors with 2.4.0 pytorch version. \
 Please run that manually before the proper fix."
@@ -213,6 +217,7 @@ def test_deploy_model(
 
         assert output.shape == (batch_size, 10)
 
+
 @pytest.mark.skip(
     reason="The test is temporarily disabled because it's causing errors with 2.4.0 pytorch version. \
 Please run that manually before the proper fix."
@@ -243,6 +248,7 @@ def test_deploy_packed_model_with_entry_point_name(
 
         assert output.shape == (batch_size, 10)
 
+
 @pytest.mark.skip(
     reason="The test is temporarily disabled because it's causing errors with 2.4.0 pytorch version. \
 Please run that manually before the proper fix."