diff --git a/concrete/numpy/np_dtypes_helpers.py b/concrete/numpy/np_dtypes_helpers.py index 8f61998cc..e974bc8ab 100644 --- a/concrete/numpy/np_dtypes_helpers.py +++ b/concrete/numpy/np_dtypes_helpers.py @@ -27,6 +27,7 @@ NUMPY_TO_COMMON_DTYPE_MAPPING: Dict[numpy.dtype, BaseDataType] = { numpy.dtype(numpy.uint64): Integer(64, is_signed=False), numpy.dtype(numpy.float32): Float(32), numpy.dtype(numpy.float64): Float(64), + numpy.dtype(bool): Integer(32, is_signed=False), } SUPPORTED_NUMPY_DTYPES = tuple(NUMPY_TO_COMMON_DTYPE_MAPPING) diff --git a/concrete/numpy/tracing.py b/concrete/numpy/tracing.py index c62068f7f..f80e8de8d 100644 --- a/concrete/numpy/tracing.py +++ b/concrete/numpy/tracing.py @@ -230,9 +230,9 @@ class NPTracer(BaseTracer): # numpy.heaviside, # numpy.hypot, # numpy.invert, - # numpy.isfinite, - # numpy.isinf, - # numpy.isnan, + numpy.isfinite, + numpy.isinf, + numpy.isnan, # numpy.isnat, # numpy.lcm, # numpy.ldexp,