added default None

Xilinx · Oct 1, 2024 · a13b41b · a13b41b
1 parent b15e2db
commit a13b41b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/core/stats/stats_wrapper.py b/src/brevitas/core/stats/stats_wrapper.py
@@ -114,7 +114,7 @@ def __init__(
         self.stats = _Stats(stats_impl, stats_output_shape)
 
     @brevitas.jit.script_method
-    def forward(self, x: Optional[torch.Tensor]) -> torch.Tensor:
+    def forward(self, x: Optional[torch.Tensor] = None) -> torch.Tensor:
         stats_input = self.first_tracked_param(x)
         if self.extra_tracked_params_list is not None:
             for extra_tracked_param in self.extra_tracked_params_list: