per-version xgboost values

rstudio · Sep 14, 2023 · d5dce53 · d5dce53
1 parent 6d33440
commit d5dce53
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/vetiver/tests/test_xgboost.py b/vetiver/tests/test_xgboost.py
@@ -64,7 +64,7 @@ def test_vetiver_build(vetiver_client):
 
     response = vetiver.predict(endpoint=vetiver_client, data=data)
 
-    assert response.iloc[0, 0] == 19.963224411010742
+    assert response.iloc[0, 0] == PREDICT_VALUE
     assert len(response) == 1
 
 
@@ -73,7 +73,7 @@ def test_batch(vetiver_client):
 
     response = vetiver.predict(endpoint=vetiver_client, data=data)
 
-    assert response.iloc[0, 0] == 19.963224411010742
+    assert response.iloc[0, 0] == PREDICT_VALUE
     assert len(response) == 3
 
 
@@ -82,7 +82,7 @@ def test_no_ptype(vetiver_client_check_ptype_false):
 
     response = vetiver.predict(endpoint=vetiver_client_check_ptype_false, data=data)
 
-    assert response.iloc[0, 0] == 19.963224411010742
+    assert response.iloc[0, 0] == PREDICT_VALUE
     assert len(response) == 1