fix test

huggingface · Jun 13, 2024 · d7734e4 · d7734e4
1 parent 3f35b3d
commit d7734e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_inference_endpoints.py b/tests/test_inference_endpoints.py
@@ -256,4 +256,4 @@ def test_resume(mock: Mock):
     endpoint = InferenceEndpoint.from_raw(MOCK_RUNNING, namespace="foo")
     mock.return_value = InferenceEndpoint.from_raw(MOCK_INITIALIZING, namespace="foo")
     endpoint.resume()
-    mock.assert_called_once_with(namespace="foo", name="my-endpoint-name", token=None)
+    mock.assert_called_once_with(namespace="foo", name="my-endpoint-name", token=None, running_ok=True)