coerce arrays to lists in random_sample()

wandb · Oct 27, 2021 · ff65af2 · ff65af2
1 parent ed2b882
commit ff65af2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/sweeps/bayes_search.py b/src/sweeps/bayes_search.py
@@ -57,6 +57,8 @@ def sigmoid(x: ArrayLike) -> ArrayLike:
 
 
 def random_sample(X_bounds: ArrayLike, num_test_samples: integer) -> ArrayLike:
+    if hasattr(X_bounds, "tolist"):
+        X_bounds = X_bounds.tolist()
     num_hyperparameters = len(X_bounds)
     test_X = np.empty((int(num_test_samples), num_hyperparameters))
     for ii in range(num_test_samples):