Skip to content

Commit 5057dd8

Browse files
authored
Merge pull request #329 from IBM/fix_master_travis
Fix tests on branch master after reverting directly pushed commit
2 parents e6d0b53 + 497f866 commit 5057dd8

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

test_requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ mxnet
88
#torch>=1.2.0
99
tensorflow
1010
scikit-learn==0.21.3
11-
xgboost
11+
xgboost==1.0.0
1212
lightgbm
1313
# Comment out because of compatibility issues with numpy versions
1414
# catboost

tests/classifiers/test_xgboost.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def setUpClass(cls):
102102

103103
def test_predict(self):
104104
y_predicted = self.classifier.predict(self.x_test[0:1])
105-
y_expected = [0.02563512, 0.02925956, 0.94510525]
105+
y_expected = [0.00280161, 0.00359648, 0.99360186]
106106
for i in range(3):
107107
self.assertAlmostEqual(y_predicted[0, i], y_expected[i], 4)
108108

tests/metrics/test_verification_decision_trees.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,8 +66,8 @@ def test_XGBoost(self):
6666
average_bound, verified_error = rt.verify(x=self.x_test, y=self.y_test, eps_init=0.3,
6767
nb_search_steps=10, max_clique=2, max_level=2)
6868

69-
self.assertEqual(average_bound, 0.035996093750000006)
70-
self.assertEqual(verified_error, 0.96)
69+
self.assertEqual(average_bound, 0.03186914062500001)
70+
self.assertEqual(verified_error, 0.99)
7171

7272
def test_LightGBM(self):
7373
train_data = lightgbm.Dataset(self.x_train, label=np.argmax(self.y_train, axis=1))

0 commit comments

Comments
 (0)