diff --git a/transformations/transformations.py b/transformations/transformations.py index 1a8b797f613181b97bd482b482c4e3070e733ff4..a1202dd75bba4125c2dd0ed2d60ed735b03ba139 100644 --- a/transformations/transformations.py +++ b/transformations/transformations.py @@ -630,10 +630,8 @@ def typeAllEquations(eqs, typeForSymbol): :return: ``fieldsRead, fieldsWritten, typedEquations`` set of read fields, set of written fields, list of equations where symbols have been replaced by typed symbols """ - if not typeForSymbol or typeForSymbol == 'double': - typeForSymbol = typingFromSympyInspection(eqs, "double") - elif typeForSymbol == 'float': - typeForSymbol = typingFromSympyInspection(eqs, "float") + if isinstance(typeForSymbol, str) or not hasattr(typeForSymbol, '__getitem__'): + typeForSymbol = typingFromSympyInspection(eqs, typeForSymbol) fieldsWritten = set() fieldsRead = set()