fix SVWP test

OnnoKampman · Feb 25, 2024 · 90498c4 · 90498c4
1 parent fb9c82b
commit 90498c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/fcest/models/test_wishart_process.py b/tests/fcest/models/test_wishart_process.py
@@ -40,7 +40,7 @@ def test_sparse_variational_wishart_process(
         k = gpflow.kernels.Matern52()
         m = SparseVariationalWishartProcess(
             D=y.shape[1],
-            Z=np.arange(y.shape[0]),
+            Z=np.arange(y.shape[0]).reshape(-1, 1),
             nu=y.shape[1],
             kernel=k,
         )