Merge pull request #79 from databricks-industry-solutions/update-glob…

…al-model-logging neuralforecast-upgrade
databricks-industry-solutions · Jan 17, 2025 · 7cb014a · 7cb014a
2 parents 8b60176 + 2e8e157
commit 7cb014a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mmf_sa/models/neuralforecast/NeuralForecastPipeline.py b/mmf_sa/models/neuralforecast/NeuralForecastPipeline.py
@@ -54,7 +54,7 @@ def register(self, model, registered_model_name: str, input_example, run_id):
             signature=signature,
             pip_requirements=[
                 "cloudpickle==2.2.1",
-                "neuralforecast==1.7.2",
+                "neuralforecast==2.0.0",
                 "git+https://github.com/databricks-industry-solutions/many-model-forecasting.git",
                 "pyspark==3.5.0",
             ],

diff --git a/requirements-global.txt b/requirements-global.txt
@@ -1,5 +1,5 @@
 -r requirements.txt
-neuralforecast==1.7.2
+neuralforecast==2.0.0
 timesfm[torch]
 chronos-forecasting
 uni2ts