From 5224c167417c0f321b916e5c8bd1900107fff4c3 Mon Sep 17 00:00:00 2001 From: Kaushik Kulkarni <kaushikcfd@gmail.com> Date: Tue, 29 Jun 2021 06:12:12 -0500 Subject: [PATCH] take advantage of the fact that super is the one that set self.np --- arraycontext/impl/pytato/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py index 68f4856..8080395 100644 --- a/arraycontext/impl/pytato/__init__.py +++ b/arraycontext/impl/pytato/__init__.py @@ -55,12 +55,14 @@ class PytatoPyOpenCLArrayContext(ArrayContext): super().__init__() self.queue = queue self.allocator = allocator - 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