From f5781d90e34e949927af2974efae3ef52ab42c72 Mon Sep 17 00:00:00 2001
From: Vincent Delaitre <vincent@deepomatic.com>
Date: Tue, 2 Apr 2024 10:15:01 +0200
Subject: [PATCH] Remove deployment task support

---
 tests/test_client.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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