fix super

Xilinx · Sep 10, 2024 · b4d744d · b4d744d
1 parent d50a311
commit b4d744d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/export/inference/handler.py b/src/brevitas/export/inference/handler.py
@@ -71,7 +71,7 @@ def prepare_for_export(self, module):
             if module._cached_weight is not None and not module.cache_inference_quant_weight_metadata_only:
                 self.cached_weight = module._cached_weight
             else:
-                super.prepare_for_export(module)
+                super().prepare_for_export(module)
 
     def forward(self, x) -> Tuple[torch.Tensor]:
         if self.cached_weight is not None: