Fixup

qubvel-org · Jan 15, 2025 · 20564f2 · 20564f2
1 parent 17a4b70
commit 20564f2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/tests/encoders/base.py b/tests/encoders/base.py
@@ -213,7 +213,9 @@ def test_compile(self):
         encoder = encoder.eval().to(default_device)
 
         torch.compiler.reset()
-        compiled_encoder = torch.compile(encoder, fullgraph=True, dynamic=True, backend="eager")
+        compiled_encoder = torch.compile(
+            encoder, fullgraph=True, dynamic=True, backend="eager"
+        )
 
         if encoder._is_torch_compilable:
             compiled_encoder(sample)

diff --git a/tests/models/base.py b/tests/models/base.py
@@ -232,7 +232,9 @@ def test_compile(self):
         model = model.eval().to(default_device)
 
         torch.compiler.reset()
-        compiled_model = torch.compile(model, fullgraph=True, dynamic=True, backend="eager")
+        compiled_model = torch.compile(
+            model, fullgraph=True, dynamic=True, backend="eager"
+        )
 
         with torch.inference_mode():
             compiled_model(sample)