diff --git a/pymbolic/traits.py b/pymbolic/traits.py index 8ac544ff25b62e12eb7a22f88cd813b5052cb649..8cbff25291d8122a0e15fddcd78eb7d92ebd7b2c 100644 --- a/pymbolic/traits.py +++ b/pymbolic/traits.py @@ -40,11 +40,12 @@ def traits(x): try: return x.traits() except AttributeError: - if isinstance(x, (complex, float)): return FieldTraits() - if isinstance(x, int): return IntegerTraits() - raise NoTraitsError - - + if isinstance(x, (complex, float)): + return FieldTraits() + elif isinstance(x, (int, long)): + return IntegerTraits() + else: + raise NoTraitsError def common_traits(*args):