fix test_squiggle_int

wandb · Oct 27, 2021 · 3d4aa22 · 3d4aa22
1 parent ed2b882
commit 3d4aa22
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/sweeps/bayes_search.py b/src/sweeps/bayes_search.py
@@ -61,8 +61,8 @@ def random_sample(X_bounds: ArrayLike, num_test_samples: integer) -> ArrayLike:
     test_X = np.empty((int(num_test_samples), num_hyperparameters))
     for ii in range(num_test_samples):
         for jj in range(num_hyperparameters):
-            if type(X_bounds[jj][0]) == int:
-                assert type(X_bounds[jj][1]) == int
+            if type(X_bounds[jj][0].tolist()) == int:
+                assert type(X_bounds[jj][1].tolist()) == int
                 test_X[ii, jj] = np.random.randint(X_bounds[jj][0], X_bounds[jj][1])
             else:
                 test_X[ii, jj] = (