There was an error while loading. Please reload this page.
1 parent af5e3d8 commit d08d0b7Copy full SHA for d08d0b7
1 file changed
tests/unit/test_batch.py
@@ -170,7 +170,6 @@ def test_batch_predict_string_warning():
170
171
mock_input = torch.tensor([[1.0], [2.0]])
172
173
- # Simulate the behavior in run_batched_loop
174
y = api.predict(mock_input)
175
with pytest.warns(
176
UserWarning,
0 commit comments