Merge pull request #520 from AnnaYasenova/patch-1

Update compat.py
This commit is contained in:
Ryan McGrath 2020-04-02 23:01:03 -07:00 committed by GitHub
commit 58587b5e07
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -9,6 +9,7 @@ Python 3 compatibility.
"""
import sys
import numpy as np
_ver = sys.version_info
@ -29,7 +30,7 @@ if is_py2:
str = unicode
basestring = basestring
numeric_types = (int, long, float)
numeric_types = (int, long, float, np.int64, np.float64)
elif is_py3:
@ -37,4 +38,4 @@ elif is_py3:
str = str
basestring = (str, bytes)
numeric_types = (int, float)
numeric_types = (int, float, np.int64, np.float64)