Skip to content
Snippets Groups Projects
Commit 55a49c90 authored by Kaushik Kulkarni's avatar Kaushik Kulkarni
Browse files

fix pt_actx stack

corrects the rec_array_multimap invocation
parent b2bd3dfa
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment