Skip to content

Commit

Permalink
Un-pin torch version
Browse files Browse the repository at this point in the history
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
  • Loading branch information
iamxiaodong authored and facebook-github-bot committed Mar 15, 2022
1 parent 0741b70 commit 8c07ac0
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 8 deletions.
7 changes: 0 additions & 7 deletions kats/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,13 +19,6 @@
from . import globalmodel # noqa
from . import metalearner # noqa
from . import model # noqa

try:
from . import neuralprophet # noqa
except ImportError:
import logging

logging.warning("kats.models.neuralprophet not available (requires neuralprophet)")
from . import nowcasting # noqa
from . import prophet # noqa
from . import quadratic_model # noqa
Expand Down
2 changes: 1 addition & 1 deletion test_requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -9,5 +9,5 @@ parameterized>=0.8.1
plotly>=2.2.1
pymannkendall>=1.4.1
pytest-mpl>=0.12
torch<=1.8.1
torch
tqdm>=4.36.1

0 comments on commit 8c07ac0

Please sign in to comment.