Merge branch 'fix-TypedImaginaryUnit.__getnewargs__' into 'master'

Add __getnewargs__ to TypedImaginaryUnit See merge request pycodegen/pystencils!94

Merge branch 'fix-TypedImaginaryUnit.getnewargs' into 'master'
Add __getnewargs__ to TypedImaginaryUnit See merge request pycodegen/pystencils!94
937c91b4 · Martin Bauer · 2f4313fd · 23463e30 · 937c91b4
Commit 937c91b4 authored 5 years ago by Martin Bauer
--- a/pystencils/data_types.py
+++ b/pystencils/data_types.py
@@ -827,3 +827,6 @@ class TypedImaginaryUnit(TypedSymbol):

    __xnew__ = staticmethod(__new_stage2__)
    __xnew_cached_ = staticmethod(cacheit(__new_stage2__))
+
+    def __getnewargs__(self):
+        return (self.dtype,)