quick lint

Signed-off-by: kvmto <kmato@nvidia.com>

Author: kvmto

code/evaluation/logical_error_rate.py

@@ -55,7 +55,10 @@
 from typing import Optional
 
 _compile_disabled = os.environ.get("PREDECODER_TORCH_COMPILE", "").strip().lower() in (
-    "0", "false", "no", "off",
+    "0",
+    "false",
+    "no",
+    "off",
 )
 
 

code/qec/surface_code/homological_equivalence_torch.py

@@ -43,7 +43,10 @@
 import torch
 
 _compile_disabled = os.environ.get("PREDECODER_TORCH_COMPILE", "").strip().lower() in (
-    "0", "false", "no", "off",
+    "0",
+    "false",
+    "no",
+    "off",
 )
 
 
@@ -823,7 +826,9 @@ def _get_compiled_seq_wr(num_layers: int):
     def _wr_fn(error_f, padded_masks, is_boundary, layer_valid):
         return _wr_seq_step_nobreak(error_f, padded_masks, is_boundary, layer_valid, nl)
 
-    compiled = torch.compile(_wr_fn, mode="reduce-overhead", fullgraph=True, disable=_compile_disabled)
+    compiled = torch.compile(
+        _wr_fn, mode="reduce-overhead", fullgraph=True, disable=_compile_disabled
+    )
     _compiled_seq_wr_cache[key] = compiled
     return compiled
 
@@ -2221,7 +2226,9 @@ def _timelike_loop(
 
         return x_work, z_work, sz_work, sx_work
 
-    compiled = torch.compile(_timelike_loop, mode="reduce-overhead", fullgraph=True, disable=_compile_disabled)
+    compiled = torch.compile(
+        _timelike_loop, mode="reduce-overhead", fullgraph=True, disable=_compile_disabled
+    )
     _compiled_timelike_cache[key] = compiled
     return compiled
 
@@ -2289,7 +2296,9 @@ def _w2_loop(
 
         return x_work, z_work, sz_work, sx_work
 
-    compiled = torch.compile(_w2_loop, mode="max-autotune", fullgraph=True, disable=_compile_disabled)
+    compiled = torch.compile(
+        _w2_loop, mode="max-autotune", fullgraph=True, disable=_compile_disabled
+    )
     _compiled_weight2_cache[key] = compiled
     return compiled