Skip to content

Commit d835f96

Browse files
Use ExecutionPlacementError from dpnp.exceptions in dpnp
1 parent d30c357 commit d835f96

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

dpnp/tests/test_fft.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -277,7 +277,7 @@ def test_validate_out(self):
277277
# Inconsistent sycl_queue
278278
a = dpnp.ones((10,), dtype=dpnp.complex64, sycl_queue=dpctl.SyclQueue())
279279
out = dpnp.empty((10,), sycl_queue=dpctl.SyclQueue())
280-
assert_raises(dpt.ExecutionPlacementError, dpnp.fft.fft, a, out=out)
280+
assert_raises(ExecutionPlacementError, dpnp.fft.fft, a, out=out)
281281

282282
# Invalid shape
283283
a = dpnp.ones((10,), dtype=dpnp.complex64)

dpnp/tests/test_sycl_queue.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -674,7 +674,7 @@ def test_2in_broadcasting(func, data1, data2, device):
674674
def test_2in_1out_diff_queue_but_equal_context(func, device):
675675
x1 = dpnp.arange(10)
676676
x2 = dpnp.arange(10, sycl_queue=dpctl.SyclQueue(device))[::-1]
677-
with assert_raises((ValueError, dpt.ExecutionPlacementError)):
677+
with assert_raises((ValueError, ExecutionPlacementError)):
678678
getattr(dpnp, func)(x1, x2)
679679

680680

0 commit comments

Comments
 (0)