Fix (minifloat): make MaxFloatInfNaN jit compatible

src/brevitas/core/function_wrapper/clamp.py

@@ -169,11 +169,13 @@ def __init__(
         if any(map(lambda x: len(x) > mantissa_bit_width, self.__special_values)):
             raise RuntimeError('NaN/inf codes need to be the same length as the mantissa.')
 
+        # move computation of min for forward pass here so it's jit compatible
+        self.__min_special_case = min(map(lambda x: int(x, 2), self.__special_values))
+
     @brevitas.jit.script_method
     def forward(self):
         # idea: take inf and nan values, select the smallest, set max_value to smallest_val - 1
-        min_special_case = min(map(lambda x: int(x, 2), self.__special_values))
-        max_value_mantissa = min_special_case - 1
+        max_value_mantissa = self.__min_special_case - 1
 
         if max_value_mantissa < 0:
             # all mantissa values are used, so we need to use decrease exponent values