From ac224c0dbd16e9fe1f8487b0fead9192c75dd48b Mon Sep 17 00:00:00 2001 From: Max Zwiessele Date: Fri, 17 Oct 2014 17:31:47 +0100 Subject: [PATCH] [kernel slicing] active_dims can be a single integer now --- GPy/kern/_src/kern.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GPy/kern/_src/kern.py b/GPy/kern/_src/kern.py index db440cab..506d32e6 100644 --- a/GPy/kern/_src/kern.py +++ b/GPy/kern/_src/kern.py @@ -49,7 +49,7 @@ class Kern(Parameterized): if active_dims is None: active_dims = np.arange(input_dim) - self.active_dims = np.array(active_dims, dtype=int) + self.active_dims = np.atleast_1d(active_dims).astype(int) assert self.active_dims.size == self.input_dim, "input_dim={} does not match len(active_dim)={}, active_dims={}".format(self.input_dim, self.active_dims.size, self.active_dims)