diff --git a/arraycontext/impl/pytato/fake_numpy.py b/arraycontext/impl/pytato/fake_numpy.py index 6c78d4195684f722b56eb8cc83623af14841567b..7b01a1371b60bb50caf6f3e6a5151a5295148001 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