We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 46aefc7 + 0f5c31c commit b4e388aCopy full SHA for b4e388a
1 file changed
vetiver/tests/test_pytorch.py
@@ -99,7 +99,7 @@ def test_torch_predict_ptype_error():
99
assert response.status_code == 422, response.text # value is not a valid float
100
101
102
-def test_torch_predict_no_ptype_error():
+def test_torch_predict_no_ptype_batch():
103
torch.manual_seed(3)
104
x_train, torch_model = _build_torch_v()
105
v = VetiverModel(torch_model, model_name = "torch")
@@ -112,7 +112,7 @@ def test_torch_predict_no_ptype_error():
112
assert response.json() == {"prediction":[[-4.060722351074219],[-4.060722351074219]]}, response.text
113
114
115
-def test_torch_predict_no_ptype_batch():
+def test_torch_predict_no_ptype():
116
117
118
0 commit comments