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

…al-model-logging pin ray dependency for neuralforecast deployment
databricks-industry-solutions · Jan 17, 2025 · 60af15e · 60af15e
2 parents c904aaf + d02150e
commit 60af15e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmf_sa/models/neuralforecast/NeuralForecastPipeline.py b/mmf_sa/models/neuralforecast/NeuralForecastPipeline.py
@@ -55,6 +55,7 @@ def register(self, model, registered_model_name: str, input_example, run_id):
             pip_requirements=[
                 "cloudpickle==2.2.1",
                 "neuralforecast==2.0.0",
+                "ray[tune] == 2.5.0",
                 "git+https://github.com/databricks-industry-solutions/many-model-forecasting.git",
                 "pyspark==3.5.0",
             ],