Looks like we need a custom __copy__ method after all

asdf-format · May 4, 2020 · f64413d · f64413d
1 parent 96d47a5
commit f64413d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/asdf/tagged.py b/asdf/tagged.py
@@ -32,7 +32,7 @@
 """
 
 from collections import UserDict, UserList, UserString
-from copy import deepcopy
+from copy import deepcopy, copy
 
 
 __all__ = ['tag_object', 'get_tag']
@@ -68,6 +68,10 @@ def __deepcopy__(self, memo):
         data_copy = deepcopy(self.data, memo)
         return TaggedDict(data_copy, self._tag)
 
+    def __copy__(self):
+        data_copy = copy(self.data)
+        return TaggedDict(data_copy, self._tag)
+
 
 class TaggedList(Tagged, UserList, list):
     """
@@ -90,6 +94,10 @@ def __deepcopy__(self, memo):
         data_copy = deepcopy(self.data, memo)
         return TaggedList(data_copy, self._tag)
 
+    def __copy__(self):
+        data_copy = copy(self.data)
+        return TaggedList(data_copy, self._tag)
+
 
 class TaggedString(Tagged, UserString, str):
     """