Skip to content

Commit

Permalink
Add autogluon src
Browse files Browse the repository at this point in the history
  • Loading branch information
Abdul Fatir Ansari committed Nov 18, 2024
1 parent ec60265 commit 23b8c84
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 1 deletion.
3 changes: 3 additions & 0 deletions src/chronos/base.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,9 @@
# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0

# Authors: Caner Turkmen <[email protected]>, Abdul Fatir Ansari <[email protected]>, Lorenzo Stella <[email protected]>
# Original source:
# https://github.com/autogluon/autogluon/blob/f57beb26cb769c6e0d484a6af2b89eab8aee73a8/timeseries/src/autogluon/timeseries/models/chronos/pipeline/base.py

from enum import Enum
from pathlib import Path
Expand Down
4 changes: 3 additions & 1 deletion src/chronos/chronos_bolt.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0

# Authors: Abdul Fatir Ansari <[email protected]>, Caner Turkmen <[email protected]>, Lorenzo Stella <[email protected]>
# Original source:
# https://github.com/autogluon/autogluon/blob/f57beb26cb769c6e0d484a6af2b89eab8aee73a8/timeseries/src/autogluon/timeseries/models/chronos/pipeline/chronos_bolt.py

import copy
import logging
Expand Down Expand Up @@ -431,7 +433,7 @@ def predict( # type: ignore[override]
Refer to the base method (``BaseChronosPipeline.predict``)
for details on shared parameters.
limit_prediction_length
Force prediction length smaller or equal than the
built-in prediction length from the model. False by
Expand Down

0 comments on commit 23b8c84

Please sign in to comment.