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