Skip to content

Commit b4e388a

Browse files
committed
Merge branch 'single-dispatch' of https://github.com/has2k1/vetiver-python into single-dispatch
2 parents 46aefc7 + 0f5c31c commit b4e388a

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

vetiver/tests/test_pytorch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ def test_torch_predict_ptype_error():
9999
assert response.status_code == 422, response.text # value is not a valid float
100100

101101

102-
def test_torch_predict_no_ptype_error():
102+
def test_torch_predict_no_ptype_batch():
103103
torch.manual_seed(3)
104104
x_train, torch_model = _build_torch_v()
105105
v = VetiverModel(torch_model, model_name = "torch")
@@ -112,7 +112,7 @@ def test_torch_predict_no_ptype_error():
112112
assert response.json() == {"prediction":[[-4.060722351074219],[-4.060722351074219]]}, response.text
113113

114114

115-
def test_torch_predict_no_ptype_batch():
115+
def test_torch_predict_no_ptype():
116116
torch.manual_seed(3)
117117
x_train, torch_model = _build_torch_v()
118118
v = VetiverModel(torch_model, model_name = "torch")

0 commit comments

Comments
 (0)