Reorganise and explicitly state what's importable

This commit is contained in:
ines 2017-05-18 14:12:31 +02:00
parent 57beef5d36
commit 33decd85b6

View File

@ -16,6 +16,11 @@ try:
except ImportError:
import copyreg as copy_reg
pickle = pickle
copy_reg = copy_reg
CudaStream = CudaStream
cupy = cupy
fix_text = ftfy.fix_text
is_python2 = six.PY2
is_python3 = six.PY3
@ -23,8 +28,6 @@ is_windows = sys.platform.startswith('win')
is_linux = sys.platform.startswith('linux')
is_osx = sys.platform == 'darwin'
fix_text = ftfy.fix_text
if is_python2:
bytes_ = str