From a1b3c5ce6dfac0481a3205cfb12b51aee8b2f9f1 Mon Sep 17 00:00:00 2001 From: Thiago Castro Ferreira Date: Wed, 13 Nov 2024 17:06:37 -0300 Subject: [PATCH] Fixing pipeline poll --- aixplain/modules/pipeline/asset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aixplain/modules/pipeline/asset.py b/aixplain/modules/pipeline/asset.py index f9e29235..308f19b3 100644 --- a/aixplain/modules/pipeline/asset.py +++ b/aixplain/modules/pipeline/asset.py @@ -139,7 +139,7 @@ def poll(self, poll_url: Text, name: Text = "pipeline_process") -> Dict: r = _request_with_retry("get", poll_url, headers=headers) try: resp = r.json() - if isinstance(resp["data"], str): + if "data" in resp and isinstance(resp["data"], str): try: resp["data"] = json.loads(resp["data"])["response"] except Exception: