precommit

Xilinx · Feb 23, 2024 · 15beb27 · 15beb27
1 parent f106c7b
commit 15beb27
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/brevitas/proxy/parameter_quant.py b/src/brevitas/proxy/parameter_quant.py
@@ -184,7 +184,9 @@ def bit_width(self):
         bit_width = self.__call__(self.tracked_parameter_list[0], zhs).bit_width
         return bit_width
 
-    def forward(self, x: Tensor, input_scale: Optional[Tensor] = None) ->  Union[Tensor, QuantTensor]:
+    def forward(self,
+                x: Tensor,
+                input_scale: Optional[Tensor] = None) -> Union[Tensor, QuantTensor]:
         if self.is_quant_enabled:
             impl = self.export_handler if self.export_mode else self.tensor_quant
             if self.requires_input_scale and input_scale is None: