DOFArray: support untaggable arrays in setstate

inducer · Aug 18, 2023 · d6e600c · d6e600c
1 parent 3664292
commit d6e600c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/meshmode/dof_array.py b/meshmode/dof_array.py
@@ -333,7 +333,12 @@ def __setstate__(self, state):
             assert len(axes_tags[idx]) == ary.ndim
             assert isinstance(axes_tags[idx], list)
 
-            d = actx.from_numpy(ary)._with_new_tags(tags[idx])
+            try:
+                d = actx.from_numpy(ary)._with_new_tags(tags[idx])
+            except AttributeError:
+                # 'actx.from_numpy' might return an array that does not have
+                # '_with_new_tags' (e.g., np.ndarray).
+                d = actx.from_numpy(ary)
 
             for ida, ax in enumerate(axes_tags[idx]):
                 d = actx.tag_axis(ida, ax, d)