From f2d09956b61fb38d193e72d6c35567dc3a6cdea0 Mon Sep 17 00:00:00 2001
From: markus holzer <markus.holzer@fau.de>
Date: Sun, 25 Apr 2021 10:21:57 +0200
Subject: [PATCH] Fixed sympy 1.9 bug partial

---
 pystencils/data_types.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pystencils/data_types.py b/pystencils/data_types.py
index 28fc4b32..4fc236e5 100644
--- a/pystencils/data_types.py
+++ b/pystencils/data_types.py
@@ -242,6 +242,9 @@ class TypedSymbol(sp.Symbol):
     def __getnewargs__(self):
         return self.name, self.dtype
 
+    def __getnewargs_ex__(self):
+        return ((self.name, self.dtype), self.assumptions0)
+
     @property
     def canonical(self):
         return self
-- 
GitLab