From c5ac133c844a07d0e5b619587d870d3e5d8c35b8 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Wed, 30 Oct 2024 14:17:03 +0000 Subject: [PATCH] return args and precommit --- src/brevitas/export/onnx/qonnx/handler.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/brevitas/export/onnx/qonnx/handler.py b/src/brevitas/export/onnx/qonnx/handler.py index 3b9a2d41c..939e3d32d 100644 --- a/src/brevitas/export/onnx/qonnx/handler.py +++ b/src/brevitas/export/onnx/qonnx/handler.py @@ -56,7 +56,8 @@ def prepare_for_export(self, module): def symbolic_execution(self, x: Tensor): x = BrevitasFloatQuantFn.apply(x, *self.symbolic_kwargs.values()) - return x, *self.return_args.values() + return_args = (x, *self.return_args.values()) + return return_args class BrevitasWeightFloatQuantProxyHandler(BrevitasFloatQuantProxyHandler):