add missing warnings module
@@ -3,6 +3,8 @@ Functionality for evaluating expressions across entire datasets.
Includes multi-GPU support for fast evaluation.
"""
+import warnings
+
from distutils.version import LooseVersion
import numpy as np
from six.moves import range
