init arrays in numpy style

apache · Jan 27, 2023 · 0189c07 · 0189c07
1 parent 6314fa3
commit 0189c07
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py
@@ -6792,12 +6792,12 @@ def get_bernoulli_model(shape, in_dtype="float32", out_dtype="int32", seed=None)
 
     # Binomial test input with 0.5 values
     val_num = 10000
-    arr = [0.5] * val_num
+    arr = np.ones([val_num]) * 0.5
     inputs = np.array(arr).astype("float32")
     verify_bernoulli(inputs)
 
     # Binomial test input with 0.1 values
-    arr = [0.1] * val_num
+    arr = np.ones([val_num]) * 0.1
     inputs = np.array(arr).astype("float32")
     verify_bernoulli(inputs)