Skip to content

Commit d5dce53

Browse files
committed
per-version xgboost values
1 parent 6d33440 commit d5dce53

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

vetiver/tests/test_xgboost.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def test_vetiver_build(vetiver_client):
6464

6565
response = vetiver.predict(endpoint=vetiver_client, data=data)
6666

67-
assert response.iloc[0, 0] == 19.963224411010742
67+
assert response.iloc[0, 0] == PREDICT_VALUE
6868
assert len(response) == 1
6969

7070

@@ -73,7 +73,7 @@ def test_batch(vetiver_client):
7373

7474
response = vetiver.predict(endpoint=vetiver_client, data=data)
7575

76-
assert response.iloc[0, 0] == 19.963224411010742
76+
assert response.iloc[0, 0] == PREDICT_VALUE
7777
assert len(response) == 3
7878

7979

@@ -82,7 +82,7 @@ def test_no_ptype(vetiver_client_check_ptype_false):
8282

8383
response = vetiver.predict(endpoint=vetiver_client_check_ptype_false, data=data)
8484

85-
assert response.iloc[0, 0] == 19.963224411010742
85+
assert response.iloc[0, 0] == PREDICT_VALUE
8686
assert len(response) == 1
8787

8888

0 commit comments

Comments
 (0)