From c5cf36e4619e00b3c52064e78999c013be5d32dc Mon Sep 17 00:00:00 2001 From: Marius Brehler Date: Thu, 2 Nov 2023 14:54:17 +0000 Subject: [PATCH] Adjust type check --- scripts/generate_testscases.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/generate_testscases.py b/scripts/generate_testscases.py index 25f83ead..0640362a 100644 --- a/scripts/generate_testscases.py +++ b/scripts/generate_testscases.py @@ -32,7 +32,7 @@ def generate_example(model_input): dtype = model_input.dtype - if dtype == np.float32: + if dtype == "float32": return np.random.uniform(low=0.0, high=1.0, size=shape).astype(np.float32) else: