From 807b7fb2ca771f522c35d55f7aa697fcba4798f0 Mon Sep 17 00:00:00 2001 From: Matthias Diener <mdiener@illinois.edu> Date: Thu, 10 Jun 2021 17:15:27 -0500 Subject: [PATCH] flake8 fixes --- arraycontext/fake_numpy.py | 1 + arraycontext/impl/pyopencl.py | 1 - arraycontext/impl/pytato.py | 3 ++- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/arraycontext/fake_numpy.py b/arraycontext/fake_numpy.py index 7e9b189..e5c281c 100644 --- a/arraycontext/fake_numpy.py +++ b/arraycontext/fake_numpy.py @@ -196,6 +196,7 @@ class BaseFakeNumpyLinalgNamespace: ord = 2 from arraycontext.impl import _is_meshmode_dofarray + from arraycontext.impl.pyopencl import _flatten_array if _is_meshmode_dofarray(ary): from warnings import warn diff --git a/arraycontext/impl/pyopencl.py b/arraycontext/impl/pyopencl.py index d0777a6..0e6f8d9 100644 --- a/arraycontext/impl/pyopencl.py +++ b/arraycontext/impl/pyopencl.py @@ -39,7 +39,6 @@ from arraycontext.metadata import FirstAxisIsElementsTag from arraycontext.fake_numpy import \ BaseFakeNumpyNamespace, BaseFakeNumpyLinalgNamespace from arraycontext.container.traversal import rec_multimap_array_container -from arraycontext.container import serialize_container from arraycontext.context import ArrayContext diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py index ebc69df..552fc9f 100644 --- a/arraycontext/impl/pytato.py +++ b/arraycontext/impl/pytato.py @@ -288,7 +288,8 @@ class PytatoArrayContext(ArrayContext): if isinstance(array, cla.Array): return array.with_queue(None) if not isinstance(array, pt.Array): - raise TypeError(f"PytatoArrayContext.freeze invoked with non-pt array of type '{type(array)}'") + raise TypeError("PytatoArrayContext.freeze invoked with non-pt " + f"array of type '{type(array)}'") prg = pt.generate_loopy(array, cl_device=self.queue.device) evt, (cl_array,) = prg(self.queue) -- GitLab