From 34efb5ae0145130297e2a8ce8a40206c2984c950 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Sat, 6 Sep 2025 20:28:06 +0000
Subject: [PATCH 1/2] Initial plan


From acc9305717b07aada10d576c404a6c7535d2cb15 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Sat, 6 Sep 2025 20:36:19 +0000
Subject: [PATCH 2/2] Fix DrsSymbol and DrsIndexed pickle compatibility with
 SymPy v1.9+

Co-authored-by: chenpeizhi <8114085+chenpeizhi@users.noreply.github.com>
---
 drudge/drs.py | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/drudge/drs.py b/drudge/drs.py
index ac92b37..c807eff 100644
--- a/drudge/drs.py
+++ b/drudge/drs.py
@@ -135,9 +135,19 @@ def __getnewargs__(self):
         """Get the arguments for __new__."""
         return None, self.name
 
+    def __getnewargs_ex__(self):
+        """Get the arguments and keyword arguments for __new__.
+        
+        This method takes precedence over __getnewargs__ in Python 3.6+
+        and is used by SymPy v1.9+. We need to override it to ensure
+        proper pickling/unpickling of DrsSymbol objects.
+        """
+        return (None, self.name), {}
+
     def __getstate__(self):
         """Get the state for pickling."""
-        return None
+        # Return a non-None state to ensure __setstate__ is called
+        return {}
 
     def __setstate__(self, state):
         """Set the state according to pickled content."""
@@ -205,10 +215,20 @@ def __getnewargs__(self):
         """Get the arguments for __new__."""
         return (None, self.base) + self.indices
 
+    def __getnewargs_ex__(self):
+        """Get the arguments and keyword arguments for __new__.
+        
+        This method takes precedence over __getnewargs__ in Python 3.6+
+        and is used by SymPy v1.9+. We need to override it to ensure
+        proper pickling/unpickling of DrsIndexed objects.
+        """
+        return ((None, self.base) + self.indices), {}
+
     def __getstate__(self):
         """Get the state for pickling."""
-        return None
-
+        # Return a non-None state to ensure __setstate__ is called
+        return {}
+        
     def __setstate__(self, state):
         """Set the state according to pickled content."""
         from .drudge import current_drudge