Codestyle fixes

nargokul · nargokul · commit 211eb1866e21 · 2025-03-12T14:44:24.000-07:00
diff --git a/tests/integ/sagemaker/experiments/helpers.py b/tests/integ/sagemaker/experiments/helpers.py
@@ -55,4 +55,4 @@ def clear_run_context():
         f" in experiment {current_run.experiment_name}."
         " Clearing context manually"
     )
-    _RunContext.drop_current_run()
+    _RunContext.drop_current_run()
diff --git a/tests/integ/sagemaker/serve/test_serve_mlflow_pytorch_flavor_happy.py b/tests/integ/sagemaker/serve/test_serve_mlflow_pytorch_flavor_happy.py
@@ -200,7 +200,7 @@ def test_happy_pytorch_sagemaker_endpoint_with_torch_serve(
         image_uri=image_uris.retrieve(
             framework="pytorch",
             region=sagemaker_session.boto_region_name,
-            version="2.2.0", # Compatible version for py310
+            version="2.2.0",  # Compatible version for py310
             image_scope="inference",
             py_version="py310",
             instance_type=cpu_instance_type,

Original file line number	Diff line number	Diff line change
`@@ -55,4 +55,4 @@ def clear_run_context():`
`55`	`55`	`f" in experiment {current_run.experiment_name}."`
`56`	`56`	`" Clearing context manually"`
`57`	`57`	`)`
`58`		`- _RunContext.drop_current_run()`
	`58`	`+ _RunContext.drop_current_run()`