From ef0914ccc5c14ccdc74af64dc1136bbd806ce6b6 Mon Sep 17 00:00:00 2001 From: Kaushik Kulkarni <kaushikcfd@gmail.com> Date: Tue, 29 Jun 2021 05:53:17 -0500 Subject: [PATCH] avoid using a method to grab the fake numpy namespace Co-authored-by: Andreas Kloeckner <inform@tiker.net> --- arraycontext/impl/pytato/__init__.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py index 8565c66..68f4856 100644 --- a/arraycontext/impl/pytato/__init__.py +++ b/arraycontext/impl/pytato/__init__.py @@ -55,15 +55,12 @@ class PytatoPyOpenCLArrayContext(ArrayContext): super().__init__() self.queue = queue self.allocator = allocator - self.np = self._get_fake_numpy_namespace() + from arraycontext.impl.pytato.fake_numpy import PytatoFakeNumpyNamespace + self.np = PytatoFakeNumpyNamespace(self) # unused, but necessary to keep the context alive self.context = self.queue.context - def _get_fake_numpy_namespace(self): - from arraycontext.impl.pytato.fake_numpy import PytatoFakeNumpyNamespace - return PytatoFakeNumpyNamespace(self) - # {{{ ArrayContext interface def clone(self): -- GitLab