Skip to content

Commit 3d6f600

Browse files
author
Abdul Fatir Ansari
committed
Add autogluon src
1 parent 67358b2 commit 3d6f600

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

src/chronos/base.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,9 @@
11
# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
22
# SPDX-License-Identifier: Apache-2.0
33

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

58
from enum import Enum
69
from pathlib import Path

src/chronos/chronos_bolt.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22
# SPDX-License-Identifier: Apache-2.0
33

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

68
import copy
79
import logging
@@ -431,7 +433,7 @@ def predict( # type: ignore[override]
431433
432434
Refer to the base method (``BaseChronosPipeline.predict``)
433435
for details on shared parameters.
434-
436+
435437
limit_prediction_length
436438
Force prediction length smaller or equal than the
437439
built-in prediction length from the model. False by

0 commit comments

Comments
 (0)