From 5a1929d18329a9a818c2f29ee16e5ad856209b43 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Wed, 13 Sep 2023 16:39:44 -0500 Subject: [PATCH] Expose pytato.arange, pytato.full --- arraycontext/impl/pytato/fake_numpy.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/arraycontext/impl/pytato/fake_numpy.py b/arraycontext/impl/pytato/fake_numpy.py index 6c78d41..7b01a13 100644 --- a/arraycontext/impl/pytato/fake_numpy.py +++ b/arraycontext/impl/pytato/fake_numpy.py @@ -100,6 +100,12 @@ class PytatoFakeNumpyNamespace(LoopyBasedFakeNumpyNamespace): return self._array_context._rec_map_container( _full_like, ary, default_scalar=fill_value) + def arange(self, *args: Any, **kwargs: Any): + return pt.arange(*args, **kwargs) + + def full(self, shape, fill_value, dtype=None): + return pt.full(shape, fill_value, dtype) + # }}} # {{{ array manipulation routines -- GitLab