fix bug in multihyp

ChicagoHAI · Mar 4, 2025 · 77b4bd6 · 77b4bd6
1 parent 393873f
commit 77b4bd6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hypothesis_agent/data_analysis_agent/inference.py b/hypothesis_agent/data_analysis_agent/inference.py
@@ -62,13 +62,13 @@ def multiple_hypotheses_batched_predict(
 
         return predictions, actual_labels
 
-    def multiple_hypotheses_run_inference_final(
+    def run_inference_final(
         self,
         data,
         hyp_bank,
         cache_seed=None,
         max_concurrent=3,
-        generate_kwargs={},
+        **generate_kwargs,
     ):
 
         num_samples = len(data)