Adjust type check

iml130 · Nov 2, 2023 · c5cf36e · c5cf36e
1 parent b7b488b
commit c5cf36e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: