diff --git a/kats/models/__init__.py b/kats/models/__init__.py index 5b0351499..1e109d77f 100644 --- a/kats/models/__init__.py +++ b/kats/models/__init__.py @@ -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 diff --git a/test_requirements.txt b/test_requirements.txt index a87069a3d..49d5cba33 100644 --- a/test_requirements.txt +++ b/test_requirements.txt @@ -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