diff --git a/arraycontext/impl/pyopencl/__init__.py b/arraycontext/impl/pyopencl/__init__.py index aa0743925e8faa20ee3a65be4c7e759ee8f447c7..788dcdbe2d7096e16d2748104b008f4bfdd38a3a 100644 --- a/arraycontext/impl/pyopencl/__init__.py +++ b/arraycontext/impl/pyopencl/__init__.py @@ -36,7 +36,6 @@ from pytools.tag import Tag from arraycontext.metadata import FirstAxisIsElementsTag from arraycontext.context import ArrayContext -from numbers import Number if TYPE_CHECKING: diff --git a/test/test_arraycontext.py b/test/test_arraycontext.py index e68914d2f97f9885b08909d0f4f5cf6506ab6f14..3e9ec2c1242cdd18af4ebe920e627159f74bc178 100644 --- a/test/test_arraycontext.py +++ b/test/test_arraycontext.py @@ -413,7 +413,9 @@ def test_dof_array_reductions_same_as_numpy(actx_factory, op): np_red = getattr(np, op)(ary) actx_red = getattr(actx.np, op)(actx.from_numpy(ary)) actx_red = actx.to_numpy(actx_red) - + + from numbers import Number + if actx._force_device_scalars: assert actx_red.shape == () else: