Fixup

Author: qubvel

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)

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)