fix: bug in OpenAIJudge.extract_prompt_parts

wandb · Oct 31, 2024 · 9a6b3bf · 9a6b3bf
1 parent 8a83d8a
commit 9a6b3bf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/hemm/metrics/vqa/judges/mmllm_judges/openai_judge.py b/hemm/metrics/vqa/judges/mmllm_judges/openai_judge.py
@@ -84,10 +84,10 @@ def extract_prompt_parts(self, prompt: str) -> List[TaggedPromptParts]:
                     adjective = adjective.replace("the ", "")
                     adjective = adjective.replace("a ", "")
                     adjective = adjective.replace("an ", "")
+                    tagged_prompt_parts.append(
+                        TaggedPromptParts(entity=chunk_np, noun=noun, adjective=adjective)
+                    )
                     break
-            tagged_prompt_parts.append(
-                TaggedPromptParts(entity=chunk_np, noun=noun, adjective=adjective)
-            )
         return tagged_prompt_parts
 
     @weave.op()