TaggableCLArray: better __repr__

inducer · Jun 27, 2024 · dcd7d14 · dcd7d14
1 parent a081f39
commit dcd7d14
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/arraycontext/impl/pyopencl/taggable_cl_array.py b/arraycontext/impl/pyopencl/taggable_cl_array.py
@@ -99,6 +99,10 @@ def __init__(self, cq, shape, dtype, order="C", allocator=None,
         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,