From a24a2215d3efb6260ea8456c7661ddc91b1a9160 Mon Sep 17 00:00:00 2001 From: Bofang Zheng Date: Wed, 18 Sep 2024 16:39:25 -0400 Subject: [PATCH] Descriptive commit message about the changes --- imgaug/imgaug.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/imgaug/imgaug.py b/imgaug/imgaug.py index 924608c22..159c84dd4 100644 --- a/imgaug/imgaug.py +++ b/imgaug/imgaug.py @@ -36,9 +36,9 @@ # to check if a dtype instance is among these dtypes, use e.g. # `dtype.type in NP_FLOAT_TYPES` do not just use `dtype in NP_FLOAT_TYPES` as # that would fail -NP_FLOAT_TYPES = set(np.sctypes["float"]) -NP_INT_TYPES = set(np.sctypes["int"]) -NP_UINT_TYPES = set(np.sctypes["uint"]) +NP_FLOAT_TYPES = set([np.float16, np.float32, np.float64]) +NP_INT_TYPES = set([np.int8, np.int16, np.int32, np.int64]) +NP_UINT_TYPES = set([np.uint8, np.uint16, np.uint32, np.uint64]) IMSHOW_BACKEND_DEFAULT = "matplotlib"