From 16847f47e6cdb933926a0e11427303f7b130bed7 Mon Sep 17 00:00:00 2001 From: Nick Fraser Date: Tue, 3 Sep 2024 15:00:54 +0100 Subject: [PATCH] Fix (proxy/runtime): Typo fix on type hint --- src/brevitas/proxy/runtime_quant.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/brevitas/proxy/runtime_quant.py b/src/brevitas/proxy/runtime_quant.py index 0093d97c6..4ec52e47c 100644 --- a/src/brevitas/proxy/runtime_quant.py +++ b/src/brevitas/proxy/runtime_quant.py @@ -206,7 +206,9 @@ def bit_width(self, force_eval=True): return self.retrieve_attribute('bit_width', force_eval) def create_quant_tensor( - self, qt_args: Tuple[Any], x: Optional[IntQuantTensor] = None) -> IntQuantTensor: + self, + qt_args: Union[Tensor, Tuple[Any]], + x: Optional[IntQuantTensor] = None) -> IntQuantTensor: if x is None: out = IntQuantTensor(*qt_args, self.is_signed, self.training)