model.py: extra check to deal with test errors

mattpitkin · Jan 10, 2019 · 80fd703 · 80fd703
1 parent 1c49083
commit 80fd703
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pymc3/model.py b/pymc3/model.py
@@ -440,8 +440,9 @@ def __init__(self, cost, grad_vars, extra_vars=None, dtype=None,
             # manually set the TensorType for length 1 arrays due to a
             # theano conversion problem
             if isinstance(var.tag.test_value, np.ndarray):
-                if len(var.tag.test_value) == 1:
-                    shared.type = theano.tensor.TensorType(var.dtype, (True,))
+                if var.tag.test_value.ndim == 1:
+                    if len(var.tag.test_value) == 1:
+                        shared.type = theano.tensor.TensorType(var.dtype, (True,))
             self._extra_vars_shared[var.name] = shared
             givens.append((var, shared))