diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py index 8565c6689125c75327b02b7b3fd454159277636e..68f4856e1926c2c9d9126e2d313129e3baf0e207 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):