Failing test shouldn't fail anymore!

neuralmagic · Jul 11, 2023 · 64b45ac · 64b45ac
1 parent e38a570
commit 64b45ac
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/tests/server/test_endpoints.py b/tests/server/test_endpoints.py
@@ -193,14 +193,8 @@ def client(self):
         yield TestClient(app)
 
     def test_static_batch_errors_on_wrong_batch_size(self, client):
-        with pytest.raises(
-            RuntimeError,
-            match=(
-                "batch size of 1 passed into pipeline is "
-                "not divisible by model batch size of 2"
-            ),
-        ):
-            client.post("/predict/static-batch", json={"sequences": "today is great"})
+        # this is okay because we can pad batches now
+        client.post("/predict/static-batch", json={"sequences": "today is great"})
 
     def test_static_batch_good_request(self, client):
         response = client.post(