Add dependencies to environment.txt

databricks · Sep 25, 2024 · 147598b · 147598b
1 parent c1f2b3c
commit 147598b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/runtime/environment.txt b/runtime/environment.txt
@@ -2,13 +2,16 @@
 # * Keep dependencies sorted.
 
 category_encoders==2.6.0
+gluonts[torch]==0.15.1
 holidays==0.28
 hyperopt==0.2.7
+lightning==2.0.1
 mlflow==2.0.1
 numpy==1.23.5
 pandas==1.5.3
 pmdarima==2.0.3
 prophet==1.1.4
 pyarrow==8.0.0
 scikit-learn==1.1.1
+torch==2.0.1+cpu
 wrapt==1.14.1