From 32dc069ed886c67c0ec44902f2ffb7318ffbf313 Mon Sep 17 00:00:00 2001
From: Alexandru Fikl <alexfikl@gmail.com>
Date: Sat, 30 Apr 2022 23:06:00 -0500
Subject: [PATCH] implement _with_new_tags for taggable.Axis
---
arraycontext/impl/pyopencl/taggable_cl_array.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/arraycontext/impl/pyopencl/taggable_cl_array.py b/arraycontext/impl/pyopencl/taggable_cl_array.py
index aa7d12d..439ca58 100644
--- a/arraycontext/impl/pyopencl/taggable_cl_array.py
+++ b/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
--
GitLab