Added support for pickling

Author: swansonk14

tap/tap.py

@@ -668,3 +668,16 @@ def __deepcopy__(self, memo: Dict[int, Any] = None) -> TapType:
             copied.__dict__[k] = deepcopy(v, memo)
 
         return copied
+
+    def __getstate__(self) -> Dict[str, Any]:
+        """Gets the state of the object for pickling."""
+        return self.as_dict()
+
+    def __setstate__(self, d: Dict[str, Any]) -> None:
+        """
+        Initializes the object with the provided dictionary of arguments for unpickling.
+
+        :param d: A dictionary of arguments.
+        """
+        self.__init__()
+        self.from_dict(d)

tests/test_integration.py

@@ -1,6 +1,7 @@
 from copy import deepcopy
 import os
 from pathlib import Path
+import pickle
 import sys
 from tempfile import TemporaryDirectory
 from typing import Any, Iterable, List, Optional, Set, Tuple
@@ -1373,6 +1374,15 @@ def configure(self) -> None:
 
         self.assertEqual(args.a, 1)
 
+
+class TestPickle(TestCase):
+    def test_pickle(self):
+        args = IntegrationDefaultTap().parse_args([])
+        pickle_str = pickle.dumps(args)
+        loaded_args: IntegrationDefaultTap = pickle.loads(pickle_str)
+        self.assertEqual(loaded_args.as_dict(), args.as_dict())
+
+
 """
 - crash if default type not supported
 - user specifying process_args