other changes

src/trainers/quantile_trainer.py

@@ -87,7 +87,10 @@ class AutoRegressiveQuantileTrainer(AutoRegressiveTrainer):
         crps_from_samples_metric = []
 
         with torch.no_grad():
+            total_samples = len(dataloader.dataset) - 96
             for _, _, idx_batch in tqdm(dataloader):
+                idx_batch = [idx for idx in idx_batch if idx < total_samples]
+
                 if len(idx_batch) == 0:
                     continue