frozen dataclass and names compatibile with protocol for PAChoice

phelps-sg · Nov 17, 2023 · 6bbf134 · 6bbf134
1 parent 3e9492a
commit 6bbf134
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llm_cooperation/experiments/principalagent.py b/llm_cooperation/experiments/principalagent.py
@@ -32,12 +32,12 @@
 from llm_cooperation.gametypes.oneshot import OneShotResults, run_experiment
 
 
-@dataclass
+@dataclass(frozen=True)
 class PAChoice:
     value: int
 
     # pylint: disable=unused-argument
-    def description(self, participant: Participant) -> str:
+    def description(self, participant_condition: Participant) -> str:
         return str(self.value)