Remove nonfunctioning GPJax

examples/reconciliation.py

@@ -18,6 +18,7 @@
 
 class NeuralProcessForecaster(Forecaster):
     """Example implementation of a forecaster."""
+
     def __init__(self):
         super().__init__()
         self._models: list = []
@@ -71,6 +72,7 @@ def get_neural_process():
                 latent_encoder=(MLP([dim] * 3), MLP([dim, dim * 2])),
             )
             return np
+
         neural_process = get_neural_process()
         return neural_process
 

reconcile/grouping.py

@@ -79,12 +79,12 @@ def summing_matrix(self):
 
     def extract_bottom_timeseries(self, y):
         """Getter for the bottom time series."""
-        return y[..., self.n_upper_timeseries:, :]
+        return y[..., self.n_upper_timeseries :, :]
 
     def upper_time_series(self, b):
         """Getter for upper time series."""
         y = self.all_timeseries(b)
-        return y[..., :self.n_upper_timeseries, :]
+        return y[..., : self.n_upper_timeseries, :]
 
     @staticmethod
     def _paste0(a, b):

reconcile/probabilistic_reconciliation.py

@@ -81,7 +81,7 @@ def lp(x):
         initial_positions = self._forecaster.posterior_predictive(
             curr_key,
             xs_test,
-        ).sample(seed=rng_key, sample_shape=(n_chains, ))
+        ).sample(seed=rng_key, sample_shape=(n_chains,))
         initial_positions = {
             "b": self._grouping.extract_bottom_timeseries(initial_positions)
         }