From 1dd299a04ae0c21577f8762a6705ea9fd2aa01c6 Mon Sep 17 00:00:00 2001
From: Chad C Williams <ccwillia@uvic.ca>
Date: Wed, 31 May 2023 11:31:21 -0700
Subject: [PATCH] Fixed test

---
 tests/test_nearest_value_sampler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_nearest_value_sampler.py b/tests/test_nearest_value_sampler.py
index ece81a2..50a2e9c 100644
--- 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