From 55a49c904dc1a1a7b66fc3906ef30f7abb202ade Mon Sep 17 00:00:00 2001 From: Kaushik Kulkarni Date: Thu, 24 Jun 2021 13:14:50 -0500 Subject: [PATCH] fix pt_actx stack corrects the rec_array_multimap invocation --- arraycontext/impl/pytato/fake_numpy.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/arraycontext/impl/pytato/fake_numpy.py b/arraycontext/impl/pytato/fake_numpy.py index 8cfa9ad..a8a72da 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 -- GitLab