Fixed test

AutoResearch · May 31, 2023 · 1dd299a · 1dd299a
1 parent 5ded76c
commit 1dd299a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_nearest_value_sampler.py b/tests/test_nearest_value_sampler.py
@@ -4,7 +4,7 @@
 def test_output_dimensions():
     #Meta-Setup
     X_allowed = np.linspace(-3, 6, 10)
-    X = np.random.choice(X_allowed,10)
+    X = np.random.choice(X_allowed,10).reshape(-1, 1)
     n = 5
 
     #Sampler