Update nemo_skills/inference/server/reward_model.py

Co-authored-by: Shubham Toshniwal <[email protected]>

nemo_skills/inference/server/reward_model.py

@@ -118,7 +118,7 @@ def process_single_prompt(prompt):
             score = 1 / (1 + math.exp(-raw_score))
             return {"reward_model_score": score}
 
-        with ThreadPoolExecutor(len(prompts)) as executor:
+        with ThreadPoolExecutor(max_workers=len(prompts)) as executor:
             for idx, prompt in enumerate(prompts):
                 futures[executor.submit(process_single_prompt, prompt)] = idx