From 21f6b92475b05c7caee3e32389c63b655790f25d Mon Sep 17 00:00:00 2001
From: Ricardo <acq11ra@sheffield.ac.uk>
Date: Wed, 2 Sep 2015 13:32:06 +0100
Subject: [PATCH] Change in _diag_ufunc with @mzwiessele

---
 GPy/util/diag.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/GPy/util/diag.py b/GPy/util/diag.py
index e7c332e2..9a3343f0 100644
--- a/GPy/util/diag.py
+++ b/GPy/util/diag.py
@@ -46,6 +46,8 @@ def offdiag_view(A, offset=0):
     return as_strided(Af[(1+offset):], shape=(A.shape[0]-1, A.shape[1]), strides=(A.strides[0] + A.itemsize, A.strides[1]))
 
 def _diag_ufunc(A,b,offset,func):
+    b = np.squeeze(b)
+    assert b.ndim <= 1, "only implemented for one dimensional arrays"
     dA = view(A, offset); func(dA,b,dA)
     return A