file theano/gpuarray/tests/test_pickle.py

theano/gpuarray/tests/test_pickle.py

@@ -12,7 +12,7 @@ from six import reraise
 
 from nose.plugins.skip import SkipTest
 from nose.tools import assert_raises
-import numpy
+import numpy as np
 
 from theano.compat import PY3
 from theano import config
@@ -75,7 +75,7 @@ def test_unpickle_gpuarray_as_numpy_ndarray_flag2():
                     reraise(SkipTest, exc_value, exc_trace)
                 raise
 
-        assert isinstance(mat, numpy.ndarray)
+        assert isinstance(mat, np.ndarray)
         assert mat[0] == -42.0
 
     finally: