Fixed shape of tensor

StatMixedML · Aug 23, 2023 · 1aefed9 · 1aefed9
1 parent d7d70ab
commit 1aefed9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_utils/test_utils.py b/tests/test_utils/test_utils.py
@@ -20,7 +20,7 @@ def response_fn(self, request):
 
     def test_response_fn(self, response_fn):
         # Create Data for testing
-        predt = torch.tensor([1.0, 2.0, 3.0, 4.0])
+        predt = torch.tensor([1.0, 2.0, 3.0, 4.0]).reshape(-1,1)
 
         # Call the function
         predt_transformed = response_fn(predt)