diff --git a/infrastructure/abri/abri_client.py b/infrastructure/abri/abri_client.py index 302ec54f8..e1049bd0e 100644 --- a/infrastructure/abri/abri_client.py +++ b/infrastructure/abri/abri_client.py @@ -53,7 +53,9 @@ class AbriClient: return self._parse_rejection(root) job_logged = root.find("Jobs/Job_logged") if job_logged is None: - raise ValueError("Job_logged element missing from relay response") + raise AbriResponseParseError( + "Job_logged element missing from relay response" + ) return JobLogged( job_no=job_logged.attrib["job_no"], logged_info=job_logged.attrib["logged_info"], @@ -65,5 +67,5 @@ class AbriClient: code = root.findtext("code") message = root.findtext("message") if success != "false" or code is None or message is None: - raise ValueError("malformed relay failure response") + raise AbriResponseParseError("malformed relay failure response") return AbriRequestRejected(code=code, message=message)