target and dev fix

apache · Jan 19, 2023 · 63c2bc9 · 63c2bc9
1 parent 8ef65a9
commit 63c2bc9
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
@@ -6673,8 +6673,8 @@ def verify_bernoulli_with_ort(
         out_dtype="int32",
         seed=None,
         out_shape=None,
-        target=None,
-        dev=None,
+        target=target,
+        dev=dev,
         use_vm=False,
         opset=None,
         freeze_params=False,