implement _with_new_tags for taggable.Axis

arraycontext/impl/pyopencl/taggable_cl_array.py

@@ -19,9 +19,9 @@ class Axis(Taggable):
     """
     tags: FrozenSet[Tag]
 
-    def copy(self, **kwargs):
+    def _with_new_tags(self, tags: FrozenSet[Tag]) -> "Axis":
         from dataclasses import replace
-        return replace(self, **kwargs)
+        return replace(self, tags=tags)
 
 
 @memoize