assign to a results variable regardless of whether logprobs are ena…

…bled
defog-ai · Jun 24, 2024 · 49249fd · 49249fd
1 parent 7012efd
commit 49249fd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/eval/api_runner.py b/eval/api_runner.py
@@ -282,11 +282,12 @@ def run_api_eval(args):
         if not os.path.exists(output_dir):
             os.makedirs(output_dir)
 
+        results = output_df.to_dict("records")
+
         if logprobs:
             print(
                 f"Writing logprobs to JSON file at eval-visualizer/public/{output_file.split('/')[-1].replace('.csv', '.json')}"
             )
-            results = output_df.to_dict("records")
             with open(
                 f"./eval-visualizer/public/{output_file.split('/')[-1].replace('.csv', '.json')}",
                 "w",