Fix (tests): disable JIT with Mock

Xilinx · May 6, 2024 · 5fc4fb1 · 5fc4fb1
1 parent a422d0e
commit 5fc4fb1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/brevitas/core/test_float_quant.py b/tests/brevitas/core/test_float_quant.py
@@ -61,6 +61,7 @@ def test_minifloat(minifloat_format):
 
 
 @given(inp=float_tensor_random_shape_st(), minifloat_format=random_minifloat_format())
+@jit_disabled_for_mock()
 def test_float_to_quant_float(inp, minifloat_format):
     bit_width, exponent_bit_width, mantissa_bit_width, signed, exponent_bias = minifloat_format