Catch encord request error (#459)

encord-team · Nov 30, 2023 · 96b231c · 96b231c
1 parent d39687d
commit 96b231c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/encord/client.py b/encord/client.py
@@ -658,7 +658,7 @@ def add_private_data_to_dataset_get_result(
                     print("Processed all files, dataset data linking and task creation is performed, please wait")
 
                 failed_requests_count = 0
-            except requests.exceptions.RequestException:
+            except (requests.exceptions.RequestException, encord.exceptions.RequestException):
                 failed_requests_count += 1
 
                 if failed_requests_count >= LONG_POLLING_RESPONSE_RETRY_N: