Reproduce error

Author: Pushkar-Bhuse

forte/data/data_store.py

@@ -275,7 +275,8 @@ def __getstate__(self):
         state["_DataStore__elements"] = {}
 
         # Make a copy of the updated type_attributes
-        state["_type_attributes"] = deepcopy(DataStore._type_attributes)
+        type_attributes = deepcopy(DataStore._type_attributes)
+        state["_type_attributes"] = DataStore._type_attributes
 
         for k in self.__elements:
             # build the full `_type_attributes`
@@ -295,6 +296,7 @@ def __getstate__(self):
         for _, v in state["fields"].items():
             if constants.PARENT_CLASS_KEY in v:
                 v.pop(constants.PARENT_CLASS_KEY)
+        DataStore._type_attributes = type_attributes
         return state
 
     def __setstate__(self, state):

tests/forte/data/data_store_test.py

@@ -335,10 +335,14 @@ def test_get_type_info(self):
         )
         empty_data_store._get_type_info("ft.onto.base_ontology.Sentence")
         self.assertEqual(len(empty_data_store._DataStore__elements), 0)
-        self.assertEqual(
-            DataStore._type_attributes["ft.onto.base_ontology.Sentence"],
-            self.reference_type_attributes["ft.onto.base_ontology.Sentence"],
-        )
+        try:
+            self.assertEqual(
+                DataStore._type_attributes["ft.onto.base_ontology.Sentence"],
+                self.reference_type_attributes["ft.onto.base_ontology.Sentence"],
+            )
+        except AssertionError:
+            print(DataStore._type_attributes["ft.onto.base_ontology.Sentence"])
+            print(self.reference_type_attributes["ft.onto.base_ontology.Sentence"])
         self.assertEqual(
             DataStore._type_attributes["ft.onto.base_ontology.Document"],
             self.reference_type_attributes["ft.onto.base_ontology.Document"],