[kernel slicing] active_dims can be a single integer now

This commit is contained in:
Max Zwiessele 2014-10-17 17:31:47 +01:00
parent 536adf0782
commit ac224c0dbd

View file

@ -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)