diff --git a/pystencils/data_types.py b/pystencils/data_types.py index 4899bff2d4d26839796f00b6b6e49b9b0c71c9bb..af085ab3cc6ff33aa8f8e4edfd7d6b086c6d3f5a 100644 --- 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,)