diff --git a/arraycontext/impl/pytato/fake_numpy.py b/arraycontext/impl/pytato/fake_numpy.py index 8cfa9ad91013be61f3f7c9c2194e7f42fc710c8b..a8a72da3c635fc22bad5e2f8640943487b55f702 100644 --- a/arraycontext/impl/pytato/fake_numpy.py +++ b/arraycontext/impl/pytato/fake_numpy.py @@ -94,7 +94,9 @@ class PytatoFakeNumpyNamespace(BaseFakeNumpyNamespace): return pt.amax(a) def stack(self, arrays, axis=0): - return rec_multimap_array_container(pt.stack, arrays, axis) + return rec_multimap_array_container(lambda *args: pt.stack(arrays=args, + axis=axis), + *arrays) # {{{ relational operators