Fix bug in VIGP's initialization

ziatdinovmax · Feb 25, 2025 · 884f228 · 884f228
1 parent 1f7b6be
commit 884f228
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/neurobayes/models/vigp.py b/neurobayes/models/vigp.py
@@ -19,12 +19,11 @@ class VIGP(GP):
     """
 
     def __init__(self,
-                 input_dim: int,
                  kernel: kernel_fn_type,
                  priors: Optional[GPPriors] = None,
                  jitter: float = 1e-6
                  ) -> None:
-        super(VIGP, self).__init__(input_dim, kernel, priors, jitter)
+        super(VIGP, self).__init__(kernel, priors, jitter)
 
     def fit(self, X: jnp.ndarray, y: jnp.ndarray,
             num_steps: int = 1000, step_size: float = 5e-3,