From 39b68be5ed7c5e24b8f84899593dd737997d4d55 Mon Sep 17 00:00:00 2001 From: Max Zwiessele Date: Wed, 16 Apr 2014 10:17:02 +0100 Subject: [PATCH] [Kern] added assertion for inputs X being matrix (ndim == 2) --- GPy/kern/_src/kernel_slice_operations.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/GPy/kern/_src/kernel_slice_operations.py b/GPy/kern/_src/kernel_slice_operations.py index 904bb65d..10dbacee 100644 --- a/GPy/kern/_src/kernel_slice_operations.py +++ b/GPy/kern/_src/kernel_slice_operations.py @@ -31,9 +31,11 @@ class KernCallsViaSlicerMeta(ParametersChangedMeta): class _Slice_wrap(object): def __init__(self, k, X, X2=None): - assert X.ndim == 2, "only matrices are allowed as inputs to kernels for now, given X.shape={!s}".format(X.shape) self.k = k self.shape = X.shape + assert X.ndim == 2, "only matrices are allowed as inputs to kernels for now, given X.shape={!s}".format(X.shape) + if X2 is not None: + assert X2.ndim == 2, "only matrices are allowed as inputs to kernels for now, given X2.shape={!s}".format(X2.shape) if (self.k.active_dims is not None) and (self.k._sliced_X == 0): assert X.shape[1] >= len(np.r_[self.k.active_dims]), "At least {} dimensional X needed, X.shape={!s}".format(len(np.r_[self.k.active_dims]), X.shape) self.X = self.k._slice_X(X)