diff --git a/GPy/kern/_src/coregionalize.py b/GPy/kern/_src/coregionalize.py index 1381b611..3503bbd6 100644 --- a/GPy/kern/_src/coregionalize.py +++ b/GPy/kern/_src/coregionalize.py @@ -34,8 +34,8 @@ class Coregionalize(Kern): .. note: see coregionalization examples in GPy.examples.regression for some usage. """ - def __init__(self, output_dim, rank=1, W=None, kappa=None, name='coregion'): - super(Coregionalize, self).__init__(input_dim=1, name=name) + def __init__(self, input_dim, output_dim, rank=1, W=None, kappa=None, name='coregion'): + super(Coregionalize, self).__init__(input_dim, name=name) self.output_dim = output_dim self.rank = rank if self.rank>output_dim: