diff --git a/arraycontext/impl/pyopencl/fake_numpy.py b/arraycontext/impl/pyopencl/fake_numpy.py index c60f33d77dbff3763352366a71f3c25b2c7c2991..70689fc0d2326fa3795c5e6dae78a72247a862e7 100644 --- a/arraycontext/impl/pyopencl/fake_numpy.py +++ b/arraycontext/impl/pyopencl/fake_numpy.py @@ -267,11 +267,6 @@ class PyOpenCLFakeNumpyNamespace(BaseFakeNumpyNamespace): result = result.get()[()] return result - def astype(self, a, dtype): - return rec_map_array_container( - lambda x: x.astype(dtype, queue=self._array_context.queue), - a) - # }}} diff --git a/arraycontext/impl/pytato/fake_numpy.py b/arraycontext/impl/pytato/fake_numpy.py index 62b5e20086de0ec721f9a0e6724fc99d4e0ca937..9a92695587a388adfdc31050138b99e2aa267d8b 100644 --- a/arraycontext/impl/pytato/fake_numpy.py +++ b/arraycontext/impl/pytato/fake_numpy.py @@ -65,8 +65,9 @@ class PytatoFakeNumpyNamespace(BaseFakeNumpyNamespace): return super().__getattr__(name) def reshape(self, a, newshape, order="C"): - return rec_multimap_array_container( - partial(pt.reshape, order=order), a, newshape) + return rec_map_array_container( + lambda ary: pt.reshape(a, newshape, order=order), + a) def transpose(self, a, axes=None): return rec_multimap_array_container(pt.transpose, a, axes)