From bb15b4c030ace0a4b9b09e0f9feb1e6132b5198b Mon Sep 17 00:00:00 2001 From: Matthias Diener <mdiener@illinois.edu> Date: Fri, 25 Jun 2021 16:40:40 -0500 Subject: [PATCH] flake8 --- arraycontext/impl/pyopencl/__init__.py | 1 - test/test_arraycontext.py | 4 +++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/arraycontext/impl/pyopencl/__init__.py b/arraycontext/impl/pyopencl/__init__.py index aa07439..788dcdb 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 e68914d..3e9ec2c 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: -- GitLab