diff --git a/tests/test_client.py b/tests/test_client.py index d02b315..69a8c66 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -249,7 +249,9 @@ def test_create_custom_reco_and_infer(self, client, custom_network): assert data['network_id'] == custom_network['id'] assert 'post_processings' in data - client.Task.retrieve(custom_network['task_id']).wait() + # Wait for network deployment + # FIXME: it could be implemented with proper 503 error code and retry + time.sleep(30) result = spec.inference(inputs=[ImageInput(DEMO_URL)], show_discarded=False, max_predictions=3) assert inference_schema(2, 0, 'golden retriever', 0.8) == result