update dependencies

AlexanderVNikitin · Mar 21, 2024 · af53e5d · af53e5d
1 parent 0c52585
commit af53e5d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -70,8 +70,9 @@ def read_file(filename: str) -> str:
           "yfinance==0.2.28",
           "tqdm",
           "dtaidistance >= 2.3.10",
-          "tensorflow",
-          "tensorflow-probability",
+          "tensorflow < 2.16",
+          "tensorflow-probability < 0.24.0",
+          "statsmodels"
       ],
       package_data={'tsgm': ['README.md']},
       packages=find_packages())