Remove remnant of Pydantic's state in getstate (#8902)

run-llama · Nov 14, 2023 · e717f4c · e717f4c
1 parent 9c45e36
commit e717f4c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llama_index/schema.py b/llama_index/schema.py
@@ -47,8 +47,8 @@ def __getstate__(self) -> Dict[str, Any]:
         state = self.dict()
 
         # Remove common unpicklable entries
-        state["__dict__"].pop("tokenizer", None)
-        state["__dict__"].pop("tokenizer_fn", None)
+        state.pop("tokenizer", None)
+        state.pop("tokenizer_fn", None)
         return state
 
     def __setstate__(self, state: Dict[str, Any]) -> None: