Imported six to settle flake8 errors

theano/misc/cudamat_utils.py

@@ -7,6 +7,8 @@ WARNING: In the test of this file there is a transpose that is used...
          So there can be problem with shape and stride order...
 """
 
+import six
+
 try:
     import cudamat
 
@@ -16,7 +18,7 @@ try:
     if cuda.cuda_available is False:
         raise ImportError('Optional theano package cuda disabled')
 
-    if PY3:
+    if six.PY3:
         long = int
 
     def cudandarray_to_cudamat(x, copyif=False):

theano/misc/gnumpy_utils.py

@@ -2,6 +2,8 @@
 This code can only work if gnumpy and theano are initialized on the
 same gpu as theano.
 """
+
+import six
 from six.moves import reduce
 
 try:
@@ -15,7 +17,7 @@ try:
     if cuda.cuda_available is False:
         raise ImportError('Optional theano package cuda disabled')
 
-    if PY3:
+    if six.PY3:
         long = int
 
     def cudandarray_to_garray(x, copyif=False):