diff --git a/runtime/databricks/automl_runtime/forecast/prophet/forecast.py b/runtime/databricks/automl_runtime/forecast/prophet/forecast.py index 599dd38d..1b92846d 100644 --- a/runtime/databricks/automl_runtime/forecast/prophet/forecast.py +++ b/runtime/databricks/automl_runtime/forecast/prophet/forecast.py @@ -106,7 +106,7 @@ def __init__(self, horizon: int, frequency_unit: str, metric: str, interval_widt self._search_space = search_space self._algo = algo self._num_folds = num_folds - self._random_state = np.random.RandomState(random_state) + self._random_state = np.random.default_rng(random_state) self._max_eval = max_eval self._timeout = trial_timeout self._is_parallel = is_parallel diff --git a/runtime/environment.txt b/runtime/environment.txt index 5c237a44..549dcc6d 100644 --- a/runtime/environment.txt +++ b/runtime/environment.txt @@ -2,11 +2,11 @@ # * Should match package versions under https://github.com/databricks/universe/tree/master/spark/images/deps-audit/ml. # * Keep dependencies sorted. -holidays==0.10.5.2 -hyperopt==0.2.5 +holidays==0.11.2 +hyperopt==0.2.7 koalas==1.8.1 mlflow==1.20.1 -numpy==1.19.2 +numpy==1.20.2 pandas==1.2.4 prophet==1.0.1 pyarrow==4.0.0