fix missing type vars

phelps-sg · Nov 5, 2023 · 9d3d8b4 · 9d3d8b4
1 parent 5e261d4
commit 9d3d8b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llm_cooperation/gametypes/oneshot.py b/llm_cooperation/gametypes/oneshot.py
@@ -52,7 +52,7 @@ def to_df(self) -> pd.DataFrame:
 def play_game(
     role_prompt: RT,
     participant_condition: PT,
-    generate_instruction_prompt: PromptGenerator,
+    generate_instruction_prompt: PromptGenerator[PT, RT],
     model_setup: ModelSetup,
 ) -> List[Completion]:
     gpt_completions, messages = start_game(