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

OnScale · Feb 7, 2024 · b2ad171 · b2ad171
1 parent 36384dc
commit b2ad171
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: