From 884f2285fbe18663fa1e1c2408d2507607dbd173 Mon Sep 17 00:00:00 2001
From: Maxim Ziatdinov <ziatdinovmax@gmail.com>
Date: Mon, 24 Feb 2025 16:54:58 -0800
Subject: [PATCH] Fix bug in VIGP's initialization

---
 neurobayes/models/vigp.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/neurobayes/models/vigp.py b/neurobayes/models/vigp.py
index c7f0489..546c1d2 100644
--- 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,