From b4d744d2f0c672a23593e1657a45eb783eaf4f45 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Tue, 10 Sep 2024 10:50:43 +0100 Subject: [PATCH] fix super --- src/brevitas/export/inference/handler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/brevitas/export/inference/handler.py b/src/brevitas/export/inference/handler.py index bd81cadc3..39a047437 100644 --- 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: