Fixed test

valence-labs · Apr 2, 2024 · b968257 · b968257
1 parent fe4b86a
commit b968257
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_regressor.py b/tests/test_regressor.py
@@ -27,4 +27,4 @@ def test_regressors(small_dummy):
         reg.solver = reg._get_solver()
         assert isinstance(reg.solver, inst)
         results = reg.solve()
-        assert results[0].shape == (99, 2)
+        assert results[1].shape == 2