Add case where 'axis' is a numpy scalar array

If the 'axis' variable is a numpy scalar array (like numpy.array(1)), these conditional branches will fail as it tries to build a set over the non-iterable array. I've added another branch to handle this case.

Add case where 'axis' is a numpy scalar array
8c8ee26e · Jeremiah Lowin · 2345e188 · 8c8ee26e
--- a/theano/tensor/elemwise.py
+++ b/theano/tensor/elemwise.py
@@ -1250,6 +1250,8 @@ class CAReduce(Op):
        # See <http://projects.scipy.org/numpy/ticket/2235>.
        elif isinstance(axis, (int, numpy.integer)):
            self.axis = (axis,)
+        elif isinstance(axis, numpy.ndarray) and axis.ndim == 0:
+            self.axis = (int(axis),)
        else:
            self.axis = list(set(axis))
            self.axis.sort()