From dcd7d14b5e840e1a25afe5e6e74a629790f34c58 Mon Sep 17 00:00:00 2001 From: Matthias Diener Date: Thu, 27 Jun 2024 15:02:10 -0500 Subject: [PATCH] TaggableCLArray: better __repr__ --- arraycontext/impl/pyopencl/taggable_cl_array.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/arraycontext/impl/pyopencl/taggable_cl_array.py b/arraycontext/impl/pyopencl/taggable_cl_array.py index 32fa6d7..af3ab15 100644 --- a/arraycontext/impl/pyopencl/taggable_cl_array.py +++ b/arraycontext/impl/pyopencl/taggable_cl_array.py @@ -99,6 +99,10 @@ class TaggableCLArray(cla.Array, Taggable): self.tags = tags self.axes = axes + def __repr__(self) -> str: + return (f"{type(self).__name__}(shape={self.shape}, dtype={self.dtype}, " + f"tags={self.tags}, axes={self.axes})") + def copy(self, queue=cla._copy_queue): ary = super().copy(queue=queue) return type(self)(None, tags=self.tags, axes=self.axes, -- GitLab