fix style

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

Author: kylesayrs

tests/test_transform/factory/test_correctness.py

@@ -94,4 +94,4 @@ def test_correctness_model(scheme_kwargs, model_apply, offload=False):
 @requires_accelerate()
 @pytest.mark.parametrize("scheme_kwargs", scheme_kwargs())
 def test_correctness_model_offload(scheme_kwargs, model_apply):
-    test_correctness_model(scheme_kwargs, model_apply, offload=True)
+    test_correctness_model(scheme_kwargs, model_apply, offload=True)

tests/test_transform/factory/test_memory.py

@@ -96,4 +96,4 @@ def test_memory_sharing_offload(scheme_kwargs):
 @pytest.mark.parametrize("scheme_kwargs", scheme_kwargs())
 def test_memory_sharing_training(scheme_kwargs):
     scheme_kwargs["requires_grad"] = True
-    test_memory_sharing(scheme_kwargs, offload=False)
+    test_memory_sharing(scheme_kwargs, offload=False)