diff --git a/tests/models/test_models.py b/tests/models/test_models.py index daa7ade..f705e6d 100644 --- a/tests/models/test_models.py +++ b/tests/models/test_models.py @@ -243,7 +243,6 @@ def test_using_level(model): "AutoLGBM", "AutoNHITS", "AutoTFT", - "PatchTST-FM", ]: # These models do not support levels yet with pytest.raises(ValueError) as excinfo: diff --git a/timecopilot/models/foundation/patchtst_fm.py b/timecopilot/models/foundation/patchtst_fm.py index 744b39f..c971443 100644 --- a/timecopilot/models/foundation/patchtst_fm.py +++ b/timecopilot/models/foundation/patchtst_fm.py @@ -226,11 +226,6 @@ def forecast( identifiers as the input DataFrame. """ freq = self._maybe_infer_freq(df, freq) - # When support for levels is added remove PatchTST-FM - # from the list of models that throw this exception in - # tests/models/test_models:test_using_level() - if level is not None: - raise ValueError("Level is not supported for patchtst-fm yet.") qc = QuantileConverter(level=level, quantiles=quantiles) dataset = TimeSeriesDataset.from_df( df,