Changed loss to new keyword in tests

aldro61 · May 30, 2017 · a1ebd1b · a1ebd1b
1 parent 7d22469
commit a1ebd1b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmit/tests/test_learner.py b/mmit/tests/test_learner.py
@@ -74,7 +74,7 @@ def test_noiseless_dataset_1_linear(self):
                        np.arange(len(y)).reshape(-1, 1),  # Good feature
                        np.zeros(len(y)).reshape(-1, 1)))  # Bad feature
 
-        estimator = MaxMarginIntervalTree(margin=0.0, loss="hinge")
+        estimator = MaxMarginIntervalTree(margin=0.0, loss="linear_hinge")
         estimator.fit(X, y)
 
         assert len(estimator.tree_.rules) == 1