autogluon.timeseries 1.4.1b20250830__py3-none-any.whl → 1.4.1b20251116__py3-none-any.whl

{autogluon.timeseries-1.4.1b20250830.dist-info → autogluon_timeseries-1.4.1b20251116.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: autogluon.timeseries
-Version: 1.4.1b20250830
+Version: 1.4.1b20251116
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -9,7 +9,6 @@ Project-URL: Documentation, https://auto.gluon.ai
 Project-URL: Bug Reports, https://github.com/autogluon/autogluon/issues
 Project-URL: Source, https://github.com/autogluon/autogluon/
 Project-URL: Contribute!, https://github.com/autogluon/autogluon/blob/master/CONTRIBUTING.md
-Platform: UNKNOWN
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Education
 Classifier: Intended Audience :: Developers
@@ -34,15 +33,15 @@ Classifier: Topic :: Scientific/Engineering :: Information Analysis
 Classifier: Topic :: Scientific/Engineering :: Image Recognition
 Requires-Python: >=3.9, <3.13
 Description-Content-Type: text/markdown
-License-File: ../LICENSE
-License-File: ../NOTICE
+License-File: LICENSE
+License-File: NOTICE
 Requires-Dist: joblib<1.7,>=1.2
 Requires-Dist: numpy<2.4.0,>=1.25.0
 Requires-Dist: scipy<1.17,>=1.5.4
 Requires-Dist: pandas<2.4.0,>=2.0.0
 Requires-Dist: torch<2.8,>=2.6
 Requires-Dist: lightning<2.8,>=2.5.1
-Requires-Dist: pytorch-lightning
+Requires-Dist: pytorch_lightning
 Requires-Dist: transformers[sentencepiece]<4.50,>=4.38.0
 Requires-Dist: accelerate<2.0,>=0.34.0
 Requires-Dist: gluonts<0.17,>=0.15.0
@@ -54,17 +53,35 @@ Requires-Dist: coreforecast<0.0.17,>=0.0.12
 Requires-Dist: fugue>=0.9.0
 Requires-Dist: tqdm<5,>=4.38
 Requires-Dist: orjson~=3.9
+Requires-Dist: chronos-forecasting<3,>=2.0.1
 Requires-Dist: tensorboard<3,>=2.9
-Requires-Dist: autogluon.core[raytune]==1.4.1b20250830
-Requires-Dist: autogluon.common==1.4.1b20250830
-Requires-Dist: autogluon.features==1.4.1b20250830
-Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.4.1b20250830
-Provides-Extra: all
+Requires-Dist: autogluon.core[raytune]==1.4.1b20251116
+Requires-Dist: autogluon.common==1.4.1b20251116
+Requires-Dist: autogluon.features==1.4.1b20251116
+Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost]==1.4.1b20251116
 Provides-Extra: tests
 Requires-Dist: pytest; extra == "tests"
 Requires-Dist: ruff>=0.0.285; extra == "tests"
 Requires-Dist: flaky<4,>=3.7; extra == "tests"
 Requires-Dist: pytest-timeout<3,>=2.1; extra == "tests"
+Provides-Extra: toto
+Requires-Dist: einops<1,>=0.7; extra == "toto"
+Requires-Dist: rotary-embedding-torch<1,>=0.8; extra == "toto"
+Provides-Extra: all
+Requires-Dist: rotary-embedding-torch<1,>=0.8; extra == "all"
+Requires-Dist: einops<1,>=0.7; extra == "all"
+Dynamic: author
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 
 
@@ -165,5 +182,3 @@ We are actively accepting code contributions to the AutoGluon project. If you ar
 ## :classical_building: License
 
 This library is licensed under the Apache 2.0 License.
-
-

autogluon_timeseries-1.4.1b20251116.dist-info/RECORD

@@ -0,0 +1,96 @@
+autogluon.timeseries-1.4.1b20251116-py3.9-nspkg.pth,sha256=kAlKxjI5mE3Pwwqphu2maN5OBQk8W8ew70e_qbI1c6A,482
+autogluon/timeseries/__init__.py,sha256=_CrLLc1fkjen7UzWoO0Os8WZoHOgvZbHKy46I8v_4k4,304
+autogluon/timeseries/learner.py,sha256=XTQgfZs5ZQf_7mWUz-CNnavewrfNy3ENwtGMRJWwwPQ,13889
+autogluon/timeseries/predictor.py,sha256=khISLnhVxTWMhE0WVCcTgm79K4Q9IuC-jHe01A9w1go,87468
+autogluon/timeseries/regressor.py,sha256=X9ItbQ0e3GyLpKqusjMls5uavqw8w53AH0tXfSFmVno,12049
+autogluon/timeseries/splitter.py,sha256=wK335v7cUAVPbo_9Bok1C6TFg0rB9SH3D031m0vn9-A,2342
+autogluon/timeseries/version.py,sha256=i1AvtCpe0Hg2dM7cUojAmubBtfodplDajXyAsahYUGg,91
+autogluon/timeseries/configs/__init__.py,sha256=wiLBwxZkDTQBJkSJ9-xz3p_yJxX0dbHe108dS1P5O6A,183
+autogluon/timeseries/configs/hyperparameter_presets.py,sha256=GbI2sd3uakWtaeaMyF7B5z_lmyfb6ToK6PZEUZTyG9w,2031
+autogluon/timeseries/configs/predictor_presets.py,sha256=B5HFHIelh91hhG0YYE5SJ7_14P7sylFAABgHX8n_53M,2712
+autogluon/timeseries/dataset/__init__.py,sha256=UvnhAN5tjgxXTHoZMQDy64YMDj4Xxa68yY7NP4vAw0o,81
+autogluon/timeseries/dataset/ts_dataframe.py,sha256=49Itgcrjej-x22HYMCXPGD2gjCTRkyHpY2H83aD9U9k,52384
+autogluon/timeseries/metrics/__init__.py,sha256=YJPXxsJ0tRDXq7p-sTZSLb0DuXMJH6sT1PgbZ3tMt30,3594
+autogluon/timeseries/metrics/abstract.py,sha256=6jbluvHXfLc_cuK1Fx0ZYle2sR4WGG6YxFQhkor46Q8,11545
+autogluon/timeseries/metrics/point.py,sha256=sS__n_Em7m4CUaBu3PNWQ_dHw1YCOHbEyC15fhytFL8,18308
+autogluon/timeseries/metrics/quantile.py,sha256=3XLKn01R2roLPZqcyAcxAIy_O89hdr0b4IKHyzRrXYA,4621
+autogluon/timeseries/metrics/utils.py,sha256=_Nz6GLbs91WhqN1PoA53wD4xEEuPIQ0juV5l9rDmkFo,970
+autogluon/timeseries/models/__init__.py,sha256=9NY9mqYaZe_7XB70M6psHARH-Lpkfroj4toUUPO9BmI,1339
+autogluon/timeseries/models/registry.py,sha256=8n7W04ql0ckNQUzKcAW7bxreLI8wTAUTymACgLklH9M,2158
+autogluon/timeseries/models/abstract/__init__.py,sha256=Htfkjjc3vo92RvyM8rIlQ0PLWt3jcrCKZES07UvCMV0,146
+autogluon/timeseries/models/abstract/abstract_timeseries_model.py,sha256=97HOi7fRPxtx8Y9hq-xdJI-kLMp6Z-8LUSvcfBjXFsM,31978
+autogluon/timeseries/models/abstract/model_trial.py,sha256=ENPg_7nsdxIvaNM0o0UShZ3x8jFlRmwRc5m0fGPC0TM,3720
+autogluon/timeseries/models/abstract/tunable.py,sha256=jA6p-FPZkMva67B-1foqvHK-1rr0IdEfp9RvGW1WS9I,7155
+autogluon/timeseries/models/autogluon_tabular/__init__.py,sha256=E5fZsdFPgVdyCVyj5bGmn_lQFlCMn2NvuRLBMcCFvhM,205
+autogluon/timeseries/models/autogluon_tabular/mlforecast.py,sha256=3p3ukQxWN4WQHKt3ocmIb_5VlZfHwWJikQYUhSbDbtE,36457
+autogluon/timeseries/models/autogluon_tabular/per_step.py,sha256=keEW7M4SIsu3hC4EFuxcrj5s7QjF9k_7NBARuMXmYgA,23329
+autogluon/timeseries/models/autogluon_tabular/transforms.py,sha256=XtxvaRsnmVF8strfvzEfWO5a_Q8p_wMyxHyglpO1R1c,2886
+autogluon/timeseries/models/autogluon_tabular/utils.py,sha256=Fn3Vu_Q0PCtEUbtNgLp1xIblg7dOdpFlF3W5kLHgruI,63
+autogluon/timeseries/models/chronos/__init__.py,sha256=wT77HzTtmQxW3sw2k0mA5Ot6PSHivX-Uvn5fjM05EU4,60
+autogluon/timeseries/models/chronos/model.py,sha256=N6tjC8gSOLcL5eX29JYcOgfxlRATGI2qtTZucCD83t8,33437
+autogluon/timeseries/models/chronos/utils.py,sha256=6y2wphSVYR1ylscSGdb3NvrTU4ZDgbx56Gluxht_j-k,14465
+autogluon/timeseries/models/ensemble/__init__.py,sha256=9fthsA6ozZoTC7A33O0hGhiHAMzcAgG206-b4PIF9Yc,1070
+autogluon/timeseries/models/ensemble/abstract.py,sha256=ePsz2lzmludxq4x_R1jjYgPvxMc0yqVRqHbU1Fq_pvo,4264
+autogluon/timeseries/models/ensemble/array_based/__init__.py,sha256=xCzFHS9YTPsC0LPfhh8mOWzUTYxXGz1RJ15ox0Wgr98,159
+autogluon/timeseries/models/ensemble/array_based/abstract.py,sha256=RC0PL4LvU7REF_FdQwqGT9TmeETNjFlHOJSBTeJrER8,10330
+autogluon/timeseries/models/ensemble/array_based/models.py,sha256=yvqWgXZU2iKxSe4J-kbEYHA3Lah8bYUG2-hdMNMlLP4,1640
+autogluon/timeseries/models/ensemble/array_based/regressor/__init__.py,sha256=Fw5m77f8Z5Y6UrgYFsK7bi1fIgLWdqzvoWZqkfAVmmY,327
+autogluon/timeseries/models/ensemble/array_based/regressor/abstract.py,sha256=cYsmZcjUg84EROimaBUI3X-EPIT4xGyEEqHfHFbiGYQ,2615
+autogluon/timeseries/models/ensemble/array_based/regressor/per_quantile_tabular.py,sha256=oJezyB0Kv2GPChF-Ym9IsyRX4N3OYcUx32hejvMVMTI,5061
+autogluon/timeseries/models/ensemble/array_based/regressor/tabular.py,sha256=2FnOiBDVgaldOnQcPD77mNzXJq7EDb1FGMpwjA3KAlE,4763
+autogluon/timeseries/models/ensemble/weighted/__init__.py,sha256=_LipTsDnYvTFmjZWsb1Vrm-eALsVVfUlF2gOpcaqE2Q,206
+autogluon/timeseries/models/ensemble/weighted/abstract.py,sha256=7vQVBK4TMBpESJ2EwnVklcljxmA2qWPQ9xpSREbtUwg,1543
+autogluon/timeseries/models/ensemble/weighted/basic.py,sha256=Kr8y0dlHRZg_q9AqBc3HIp1a5k_sXjrnQPlVi-63DCE,3066
+autogluon/timeseries/models/ensemble/weighted/greedy.py,sha256=zXJFenn1XxNNvCp4TlmIq1Dx3pUDWjKG1K3HsejmDeY,7323
+autogluon/timeseries/models/gluonts/__init__.py,sha256=YfyNYOkhhNsloA4MAavfmqKO29_q6o4lwPoV7L4_h7M,355
+autogluon/timeseries/models/gluonts/abstract.py,sha256=WKuUBy3ZF9VU87gaD9Us3c_xK2G1-XLeh1etVipf8hg,27769
+autogluon/timeseries/models/gluonts/dataset.py,sha256=wfEp5SPuB8bam7iTpX3Tf0FGdXp5vnZtpgC9G4VJ4tw,5111
+autogluon/timeseries/models/gluonts/models.py,sha256=1Z3x3-jVoae5X4cSnDIgJMvTJ9_O94aDSW8HEnBaL5k,25907
+autogluon/timeseries/models/local/__init__.py,sha256=e2UImoJhmj70E148IIObv90C_bHxgyLNk6YsS4p7pfs,701
+autogluon/timeseries/models/local/abstract_local_model.py,sha256=ASIZWBYs_cP0BwdrzHwblaNianPYcK5OqpqpiNxbxA0,11481
+autogluon/timeseries/models/local/naive.py,sha256=xur3WWhLaS9Iix_p_yfaStbr58nL5K4rV0dReTm3BQQ,7496
+autogluon/timeseries/models/local/npts.py,sha256=VRZk5tEJOIentt0tLM6lxyoU8US736nHOvhSAgagYMc,4203
+autogluon/timeseries/models/local/statsforecast.py,sha256=sZ6aEFzAyPNZX3rMULGWFht0Toapjb3EwHe5Rb76ZxA,33318
+autogluon/timeseries/models/multi_window/__init__.py,sha256=Bq7AT2Jxdd4WNqmjTdzeqgNiwn1NCyWp4tBIWaM-zfI,60
+autogluon/timeseries/models/multi_window/multi_window_model.py,sha256=PBnNhDXPJJatRIm9FXg9DXU_0ZkGSs2yvEqfaTwBVxM,12356
+autogluon/timeseries/models/toto/__init__.py,sha256=rQaVjZJV5ZsJGC0jhQ6CA4nYeXdV1KtlyDz2i2usQnY,54
+autogluon/timeseries/models/toto/dataloader.py,sha256=A5WHhnAe0J7fPo2KKG43hYLSrtUBGNweuqxMmClu3_A,3598
+autogluon/timeseries/models/toto/hf_pretrained_model.py,sha256=Q8bVUaSlQVE4xFn_v7H0h_NFTxzHiM1V17KFytc50jk,4783
+autogluon/timeseries/models/toto/model.py,sha256=3-5nR9qNqBFQLP6rNqBNlF4PBfnJHTcyjvz2GwdWwTg,8948
+autogluon/timeseries/models/toto/_internal/__init__.py,sha256=tKkiux9bD2Xu0AuVyTEx_sNOZutcluC7-d7tn7wsmec,193
+autogluon/timeseries/models/toto/_internal/dataset.py,sha256=xuAEOhoQNJGMoCxkLVLrgpdoOJuukAYbrSrnrkwFob0,6103
+autogluon/timeseries/models/toto/_internal/forecaster.py,sha256=UXiohiySn_Gs8kLheeVcVCO8qoEtYlEfMH1tukAOHsk,18520
+autogluon/timeseries/models/toto/_internal/backbone/__init__.py,sha256=hq5W62boH6HiEP8z3sHkI6_KM-Dd6TkDfWDm6DYE3J8,63
+autogluon/timeseries/models/toto/_internal/backbone/attention.py,sha256=HLUFoyqR8EqxUMT1BK-AjI4ClS8au35LcUo7Jx7Xhm0,9394
+autogluon/timeseries/models/toto/_internal/backbone/backbone.py,sha256=HUjpY2ZWed74UYKjp31erXF2ZHf3mmQMw_5_cCFeJGg,10104
+autogluon/timeseries/models/toto/_internal/backbone/distribution.py,sha256=8NXiaEVLuvjTW7L1t1RzooZFNERWv50zyLddbAwuYpo,2502
+autogluon/timeseries/models/toto/_internal/backbone/kvcache.py,sha256=QSVCrnbS2oD7wkJodZbP9XMVmrfCH6M3Zp44siF28Fg,5399
+autogluon/timeseries/models/toto/_internal/backbone/rope.py,sha256=Ghngo08DjHbwbyp6b-GXCyLeYR10dH-Y_RMOTYwIxPY,3527
+autogluon/timeseries/models/toto/_internal/backbone/scaler.py,sha256=opqyhHIZ6mPdPlrr3gA0qt9FFogIAYNDSq-P7CyQiqE,13728
+autogluon/timeseries/models/toto/_internal/backbone/transformer.py,sha256=5c-ngj4XHKlaedz1NkgdfQgqD2kUGkMn4mtGH_lTXsE,12410
+autogluon/timeseries/trainer/__init__.py,sha256=_tw3iioJfvtIV7wnjtEMv0yS8oabmCFxDnGRodYE7RI,72
+autogluon/timeseries/trainer/ensemble_composer.py,sha256=Vc8LfhGVUED70Y4DcIs3Jhpiur2EFXqVubgInixcb2I,9751
+autogluon/timeseries/trainer/model_set_builder.py,sha256=s6tozfND3lLfst6Vxa_oP_wgCmDapyCJYFmCjkEn-es,10788
+autogluon/timeseries/trainer/prediction_cache.py,sha256=Vi6EbMiMheq_smA93U_MoMxYUV85RdPm0dvJFdsM8K4,5551
+autogluon/timeseries/trainer/trainer.py,sha256=yAHbpTjGKzVBepzepKuXEF5SvCQXDbsnyURV6mKLqaU,52002
+autogluon/timeseries/trainer/utils.py,sha256=_hSAWOYRZsp1qX2J6pJSxLrAAWwhVROc4_cvtfiTRzU,625
+autogluon/timeseries/transforms/__init__.py,sha256=fKlT4pkJ_8Gl7IUTc3uSDzt2Xow5iH5w6fPB3ePNrTg,127
+autogluon/timeseries/transforms/covariate_scaler.py,sha256=8E5DDRLUQ3SCNDR2Yw8FZDx7DnWVdokKhNNxbp_S-9I,7017
+autogluon/timeseries/transforms/target_scaler.py,sha256=tucfrWuXwTGv0WcJMo0bSk6--CkqGMDxiFPiUFl0RB8,6142
+autogluon/timeseries/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+autogluon/timeseries/utils/features.py,sha256=GpemZRV7QiFRjZwP6NqpCVBg6m3KGBgp-eWUFzcpx54,22714
+autogluon/timeseries/utils/forecast.py,sha256=y3VV1rVCxOuh_p-2U9ftT_I5oU4gQQovxlw14jRGwyM,2259
+autogluon/timeseries/utils/warning_filters.py,sha256=SroNhLU3kwbD8anM58vdxWq36Z8j_uiY42mEt0ya-JI,2589
+autogluon/timeseries/utils/datetime/__init__.py,sha256=bTMR8jLh1LW55vHjbOr1zvWRMF_PqbvxpS-cUcNIDWI,173
+autogluon/timeseries/utils/datetime/base.py,sha256=3NdsH3NDq4cVAOSoy3XpaNixyNlbjy4DJ_YYOGuu9x4,1341
+autogluon/timeseries/utils/datetime/lags.py,sha256=rjJtdBU0M41R1jwfmvCbo045s-6XBjhGVnGBQJ9-U1E,5997
+autogluon/timeseries/utils/datetime/seasonality.py,sha256=YK_2k8hvYIMW-sJPnjGWRtCnvIOthwA2hATB3nwVoD4,834
+autogluon/timeseries/utils/datetime/time_features.py,sha256=kEOFls4Nzh8nO0Pcz1DwLsC_NA3hMI4JUlZI3kuvuts,2666
+autogluon_timeseries-1.4.1b20251116.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon_timeseries-1.4.1b20251116.dist-info/licenses/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon_timeseries-1.4.1b20251116.dist-info/METADATA,sha256=2t9sW-KP5XI6x7WUxIkqXQqBCKRruP017VYThdeOPoA,12980
+autogluon_timeseries-1.4.1b20251116.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+autogluon_timeseries-1.4.1b20251116.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon_timeseries-1.4.1b20251116.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon_timeseries-1.4.1b20251116.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon_timeseries-1.4.1b20251116.dist-info/RECORD,,

{autogluon.timeseries-1.4.1b20250830.dist-info → autogluon_timeseries-1.4.1b20251116.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.45.1)
+Generator: setuptools (79.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

autogluon/timeseries/evaluator.py

@@ -1,6 +0,0 @@
-class TimeSeriesEvaluator:
-    def __init__(self, *args, **kwargs):
-        raise ValueError(
-            "`TimeSeriesEvaluator` has been deprecated. "
-            "Please use the metrics defined in `autogluon.timeseries.metrics` instead."
-        )

autogluon/timeseries/models/chronos/pipeline/__init__.py

@@ -1,10 +0,0 @@
-from .base import BaseChronosPipeline, ForecastType
-from .chronos import ChronosPipeline
-from .chronos_bolt import ChronosBoltPipeline
-
-__all__ = [
-    "BaseChronosPipeline",
-    "ChronosBoltPipeline",
-    "ChronosPipeline",
-    "ForecastType",
-]

autogluon/timeseries/models/chronos/pipeline/base.py

@@ -1,160 +0,0 @@
-# Authors: Lorenzo Stella <stellalo@amazon.com>, Caner Turkmen <atturkm@amazon.com>
-
-from enum import Enum
-from pathlib import Path
-from typing import TYPE_CHECKING, Optional, Union
-
-import torch
-
-from .utils import left_pad_and_stack_1D
-
-if TYPE_CHECKING:
-    from transformers import PreTrainedModel
-
-
-class ForecastType(Enum):
-    SAMPLES = "samples"
-    QUANTILES = "quantiles"
-
-
-class PipelineRegistry(type):
-    REGISTRY: dict[str, "PipelineRegistry"] = {}
-
-    def __new__(cls, name, bases, attrs):
-        """See, https://github.com/faif/python-patterns."""
-        new_cls = type.__new__(cls, name, bases, attrs)
-        if name is not None:
-            cls.REGISTRY[name] = new_cls
-        if aliases := attrs.get("_aliases"):
-            for alias in aliases:
-                cls.REGISTRY[alias] = new_cls
-        return new_cls
-
-
-class BaseChronosPipeline(metaclass=PipelineRegistry):
-    forecast_type: ForecastType
-    dtypes = {
-        "bfloat16": torch.bfloat16,
-        "float32": torch.float32,
-        "float64": torch.float64,
-    }
-
-    def __init__(self, inner_model: "PreTrainedModel"):
-        """
-        Parameters
-        ----------
-        inner_model
-            A hugging-face transformers PreTrainedModel, e.g., T5ForConditionalGeneration
-        """
-        # for easy access to the inner HF-style model
-        self.inner_model = inner_model
-
-    def _prepare_and_validate_context(self, context: Union[torch.Tensor, list[torch.Tensor]]):
-        if isinstance(context, list):
-            context = left_pad_and_stack_1D(context)
-        assert isinstance(context, torch.Tensor)
-        if context.ndim == 1:
-            context = context.unsqueeze(0)
-        assert context.ndim == 2
-
-        return context
-
-    def predict(
-        self,
-        context: Union[torch.Tensor, list[torch.Tensor]],
-        prediction_length: Optional[int] = None,
-        **kwargs,
-    ):
-        """
-        Get forecasts for the given time series.
-
-        Parameters
-        ----------
-        context
-            Input series. This is either a 1D tensor, or a list
-            of 1D tensors, or a 2D tensor whose first dimension
-            is batch. In the latter case, use left-padding with
-            ``torch.nan`` to align series of different lengths.
-        prediction_length
-            Time steps to predict. Defaults to a model-dependent
-            value if not given.
-
-        Returns
-        -------
-        forecasts
-            Tensor containing forecasts. The layout and meaning
-            of the forecasts values depends on ``self.forecast_type``.
-        """
-        raise NotImplementedError()
-
-    def predict_quantiles(
-        self, context: torch.Tensor, prediction_length: int, quantile_levels: list[float], **kwargs
-    ) -> tuple[torch.Tensor, torch.Tensor]:
-        """
-        Get quantile and mean forecasts for given time series. All
-        predictions are returned on the CPU.
-
-        Parameters
-        ----------
-        context
-            Input series. This is either a 1D tensor, or a list
-            of 1D tensors, or a 2D tensor whose first dimension
-            is batch. In the latter case, use left-padding with
-            ``torch.nan`` to align series of different lengths.
-        prediction_length
-            Time steps to predict. Defaults to a model-dependent
-            value if not given.
-        quantile_levels
-            Quantile levels to compute
-
-        Returns
-        -------
-        quantiles
-            Tensor containing quantile forecasts. Shape
-            (batch_size, prediction_length, num_quantiles)
-        mean
-            Tensor containing mean (point) forecasts. Shape
-            (batch_size, prediction_length)
-        """
-        raise NotImplementedError()
-
-    @classmethod
-    def from_pretrained(
-        cls,
-        pretrained_model_name_or_path: Union[str, Path],
-        *model_args,
-        force=False,
-        **kwargs,
-    ):
-        """
-        Load the model, either from a local path or from the HuggingFace Hub.
-        Supports the same arguments as ``AutoConfig`` and ``AutoModel``
-        from ``transformers``.
-
-        When a local path is provided, supports both a folder or a .tar.gz archive.
-        """
-        from transformers import AutoConfig
-
-        kwargs.setdefault("resume_download", None)  # silence huggingface_hub warning
-        if str(pretrained_model_name_or_path).startswith("s3://"):
-            from .utils import cache_model_from_s3
-
-            local_model_path = cache_model_from_s3(str(pretrained_model_name_or_path), force=force)
-            return cls.from_pretrained(local_model_path, *model_args, **kwargs)
-
-        torch_dtype = kwargs.get("torch_dtype", "auto")
-        if torch_dtype != "auto" and isinstance(torch_dtype, str):
-            kwargs["torch_dtype"] = cls.dtypes[torch_dtype]
-
-        config = AutoConfig.from_pretrained(pretrained_model_name_or_path, **kwargs)
-        is_valid_config = hasattr(config, "chronos_pipeline_class") or hasattr(config, "chronos_config")
-
-        if not is_valid_config:
-            raise ValueError("Not a Chronos config file")
-
-        pipeline_class_name = getattr(config, "chronos_pipeline_class", "ChronosPipeline")
-        class_: Optional[BaseChronosPipeline] = PipelineRegistry.REGISTRY.get(pipeline_class_name)  # type: ignore
-        if class_ is None:
-            raise ValueError(f"Trying to load unknown pipeline class: {pipeline_class_name}")
-
-        return class_.from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)