From 3edee43865a5eed6b21c8c2bcb1309c3cbd99769 Mon Sep 17 00:00:00 2001 From: sanskarmodi8 Date: Wed, 1 Oct 2025 09:53:58 +0530 Subject: [PATCH 1/2] standardize output format of xLSTMTime estimator for point predictions --- pytorch_forecasting/models/xlstm/_xlstm.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_forecasting/models/xlstm/_xlstm.py b/pytorch_forecasting/models/xlstm/_xlstm.py index c304c67a8..dec484c23 100644 --- a/pytorch_forecasting/models/xlstm/_xlstm.py +++ b/pytorch_forecasting/models/xlstm/_xlstm.py @@ -151,6 +151,7 @@ def forward( output = output.transpose(1, 2) output = output[0, ..., : self.hparams.output_size] + output = output.unsqueeze(-1) return self.to_network_output(prediction=output) @classmethod From d629938c486932c278b501ad6cbc99a1158796dc Mon Sep 17 00:00:00 2001 From: sanskarmodi8 Date: Wed, 1 Oct 2025 10:11:01 +0530 Subject: [PATCH 2/2] removed test skip tag --- pytorch_forecasting/models/xlstm/_xlstm_pkg.py | 1 - 1 file changed, 1 deletion(-) diff --git a/pytorch_forecasting/models/xlstm/_xlstm_pkg.py b/pytorch_forecasting/models/xlstm/_xlstm_pkg.py index 94934046b..1a10fe660 100644 --- a/pytorch_forecasting/models/xlstm/_xlstm_pkg.py +++ b/pytorch_forecasting/models/xlstm/_xlstm_pkg.py @@ -17,7 +17,6 @@ class xLSTMTime_pkg(_BasePtForecaster): "capability:pred_int": False, "capability:flexible_history_length": True, "capability:cold_start": False, - "tests:skip_by_name": "test_integration", } @classmethod