Weights and y should be the same shape

ATOMScience-org · Dec 13, 2024 · a7ed96a · a7ed96a
1 parent 9c8abc3
commit a7ed96a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/atomsci/ddm/test/unit/test_transformers.py b/atomsci/ddm/test/unit/test_transformers.py
@@ -35,7 +35,7 @@ def test_all_missing_values():
 
 def test_one_task_no_missing_values():
     y = np.array([[1.0], [3.0], [5.0]])
-    w = np.array([[1, 1], [1, 1], [1, 1]])
+    w = np.array([[1], [1], [1]])
     x = np.ones_like(y)
     ids = np.array(range(len(y)))
     dataset = NumpyDataset(X=x, y=y, w=w, ids=ids)