From cb78432250759d1f479862d23d9cad58b7e4dfd4 Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Fri, 24 May 2024 13:31:59 -0500
Subject: [PATCH] Numpy actx: add arange, linspace
---
arraycontext/impl/numpy/fake_numpy.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/arraycontext/impl/numpy/fake_numpy.py b/arraycontext/impl/numpy/fake_numpy.py
index 54867c8..4ac1005 100644
--- a/arraycontext/impl/numpy/fake_numpy.py
+++ b/arraycontext/impl/numpy/fake_numpy.py
@@ -140,4 +140,10 @@ class NumpyFakeNumpyNamespace(BaseFakeNumpyNamespace):
np.logical_and),
self.array_equal, a, b)
+ def arange(self, *args, **kwargs):
+ return np.arange(*args, **kwargs)
+
+ def linspace(self, *args, **kwargs):
+ return np.linspace(*args, **kwargs)
+
# vim: fdm=marker
--
GitLab