fix random input in hyperopt (#17)

databricks · Dec 1, 2021 · 30876df · 30876df
1 parent f00e808
commit 30876df
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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