From 4317ddc8a934cddb4367c9a73ceb6700ad8a19ae Mon Sep 17 00:00:00 2001 From: bkb2135 Date: Thu, 20 Feb 2025 12:05:30 +0000 Subject: [PATCH] Check is website is dict --- prompting/rewards/web_retrieval.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/prompting/rewards/web_retrieval.py b/prompting/rewards/web_retrieval.py index 21dcec8ff..6de4242fc 100644 --- a/prompting/rewards/web_retrieval.py +++ b/prompting/rewards/web_retrieval.py @@ -217,6 +217,8 @@ def _parse_response(completion: str) -> tuple[str | None, ...]: if not isinstance(data, list) and isinstance(data, dict): data = [data] for website in data: + if not isinstance(website, dict): + continue response_url = website.get("url") response_content = website.get("content") response_relevant = website.get("relevant")