Add explicit deep copy support

lsst · Jun 21, 2023 · fbbf157 · fbbf157
1 parent d67b968
commit fbbf157
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/python/lsst/pipe/base/_quantumContext.py b/python/lsst/pipe/base/_quantumContext.py
@@ -66,6 +66,10 @@ def __post_init__(self) -> None:
                 max_mem = max_mem.to(u.B)
             object.__setattr__(self, "max_mem", max_mem)
 
+    def __deepcopy__(self, memo: Any) -> ExecutionResources:
+        """Deep copy returns itself because the class is frozen."""
+        return self
+
     def _reduce_kwargs(self) -> dict[str, Any]:
         """Return a dict of the keyword arguments that should be used
         by `__reduce__`.

diff --git a/tests/test_pipelineTask.py b/tests/test_pipelineTask.py
@@ -22,6 +22,7 @@
 """Simple unit test for PipelineTask.
 """
 
+import copy
 import pickle
 import unittest
 from typing import Any
@@ -337,6 +338,8 @@ def test_ExecutionResources(self):
         self.assertEqual(res.max_mem.value, 512)
         self.assertEqual(pickle.loads(pickle.dumps(res)), res)
 
+        self.assertIs(res, copy.deepcopy(res))
+
         with self.assertRaises(u.UnitConversionError):
             pipeBase.ExecutionResources(max_mem=1 * u.m)