fix

Xilinx · Sep 23, 2024 · 5902061 · 5902061
1 parent 50e6893
commit 5902061
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/brevitas/core/test_float_quant.py b/tests/brevitas/core/test_float_quant.py
@@ -99,7 +99,7 @@ def test_float_to_quant_float(inp, minifloat_format):
             float_clamp_impl=float_clamp)
         expected_out, *_ = float_quant(inp)
         scale = float_quant.scaling_impl(inp)
-        out_quant = float_quant.quantize(inp, scale)
+        out_quant, scale = float_quant.quantize(inp, scale)
         exponent_bit_width, mantissa_bit_width, exponent_bias  = torch.tensor(exponent_bit_width, dtype=torch.float), torch.tensor(mantissa_bit_width, dtype=torch.float), torch.tensor(exponent_bias, dtype=torch.float)
         out_quant, *_ = float_quant.float_clamp_impl(
             out_quant, exponent_bit_width, mantissa_bit_width, exponent_bias)