precommit

Xilinx · Nov 5, 2024 · ae21bca · ae21bca
1 parent 228f983
commit ae21bca
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/brevitas/core/scaling/runtime.py b/src/brevitas/core/scaling/runtime.py
@@ -60,7 +60,9 @@ def __init__(
 
     @brevitas.jit.script_method
     def forward(
-            self, x: Optional[torch.Tensor], threshold: Optional[torch.Tensor] = None) -> torch.Tensor:
+            self,
+            x: Optional[torch.Tensor],
+            threshold: Optional[torch.Tensor] = None) -> torch.Tensor:
         stats = self.parameter_list_stats(x)
         if threshold is None:
             threshold = torch.ones(1).type_as(stats)