diff --git a/torch_geometric/nn/nlp/llm_judge.py b/torch_geometric/nn/nlp/llm_judge.py index c55696e05df7..3de399a9734c 100644 --- a/torch_geometric/nn/nlp/llm_judge.py +++ b/torch_geometric/nn/nlp/llm_judge.py @@ -75,6 +75,7 @@ def _process_score(self, response: str) -> float: This is because LLMs don't always follow instructions. Credit to Gilberto. """ + print("Grading Response:", response) for i in [4, 3, 2, 1, 0]: if str(i) in response: return i / 4