Skip to content

Commit 8c07ac0

Browse files
iamxiaodongfacebook-github-bot
authored andcommitted
Un-pin torch version
Summary: We released the restriction of torch version due to the version requirement conflicts with ax package. We also removed some redundant code due to the removal of neural prophet model Reviewed By: yangbk560 Differential Revision: D34878121 fbshipit-source-id: 1878e81b338164f16c3e180b899507f928632177
1 parent 0741b70 commit 8c07ac0

File tree

2 files changed

+1
-8
lines changed

2 files changed

+1
-8
lines changed

kats/models/__init__.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -19,13 +19,6 @@
1919
from . import globalmodel # noqa
2020
from . import metalearner # noqa
2121
from . import model # noqa
22-
23-
try:
24-
from . import neuralprophet # noqa
25-
except ImportError:
26-
import logging
27-
28-
logging.warning("kats.models.neuralprophet not available (requires neuralprophet)")
2922
from . import nowcasting # noqa
3023
from . import prophet # noqa
3124
from . import quadratic_model # noqa

test_requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,5 +9,5 @@ parameterized>=0.8.1
99
plotly>=2.2.1
1010
pymannkendall>=1.4.1
1111
pytest-mpl>=0.12
12-
torch<=1.8.1
12+
torch
1313
tqdm>=4.36.1

0 commit comments

Comments
 (0)