remove mixed group

Author: phelps-sg

llm_cooperation/__init__.py

@@ -80,7 +80,6 @@ class Group(Enum):
     Competitive = "Competitive"
     Altruistic = "Altruistic"
     Selfish = "Selfish"
-    Mixed = "Mixed"
     Control = "Control"
 
 
@@ -110,10 +109,6 @@ class ModelSetup:
 CT = TypeVar("CT", bound=Choice)
 CT_co = TypeVar("CT_co", bound=Choice, covariant=True)
 CT_contra = TypeVar("CT_contra", bound=Choice, contravariant=True)
-
-# RT = TypeVar("RT")
-# RT_contra = TypeVar("RT_contra", contravariant=True)
-
 Payoffs = Tuple[float, float]
 
 

llm_cooperation/experiments/__init__.py

@@ -106,17 +106,6 @@ def all_values(enum_type: Type[Enum]) -> List[ConfigValue]:
         " always keeping an eye on your personal interests and objectives.",
     ],
     #
-    Group.Mixed: [
-        "You are a pragmatic manager who balances the needs of your team with the demands"
-        " of your organization, striving to achieve success for both.",
-        #
-        "You are a resourceful scientist who is passionate about your research but also"
-        " keenly aware of the need to secure funding and recognition for your work.",
-        #
-        "You are an ethical investor who seeks to grow your wealth while remaining committed to"
-        " sustainable and socially responsible practices.",
-    ],
-    #
     Group.Control: [
         "You are a participant in a psychology experiment.",
         #

llm_cooperation/experiments/dilemma.py

@@ -110,6 +110,16 @@ class Pronoun(Enum):
     THEY = "they"
 
 
+PD_ATTRIBUTES: Grid = {
+    CONDITION_CHAIN_OF_THOUGHT: [True, False],
+    CONDITION_LABEL: all_values(Label),
+    CONDITION_CASE: all_values(Case),
+    CONDITION_PRONOUN: all_values(Pronoun),
+    CONDITION_DEFECT_FIRST: [True, False],
+    CONDITION_LABELS_REVERSED: [True, False],
+}
+
+
 def labels(participant: Participant) -> Tuple[str, str]:
     value = participant[CONDITION_LABEL]
     result: Optional[List[str]] = None
@@ -299,24 +309,16 @@ def compute_freq_pd(choices: List[Choices[DilemmaChoice]]) -> float:
 
 @lru_cache()
 def get_participants(num_participant_samples: int) -> List[Participant]:
-    pd_attributes: Grid = {
-        CONDITION_CHAIN_OF_THOUGHT: [True, False],
-        CONDITION_LABEL: all_values(Label),
-        CONDITION_CASE: all_values(Case),
-        CONDITION_PRONOUN: all_values(Pronoun),
-        CONDITION_DEFECT_FIRST: [True, False],
-        CONDITION_LABELS_REVERSED: [True, False],
-    }
     result = list(
         participants(
             GROUP_PROMPT_CONDITIONS,
-            pd_attributes,
+            PD_ATTRIBUTES,
             num_participant_samples,
             seed=SEED_VALUE,
         )
         if num_participant_samples > 0
         else participants(
-            GROUP_PROMPT_CONDITIONS | pd_attributes,
+            GROUP_PROMPT_CONDITIONS | PD_ATTRIBUTES,
         )
     )
     for i, participant in enumerate(result):

tests/test_dilemma.py

@@ -34,6 +34,7 @@
 from llm_cooperation.experiments.dilemma import (
     CONDITION_LABELS_REVERSED,
     CONDITION_PRONOUN,
+    PD_ATTRIBUTES,
     Cooperate,
     Defect,
     DilemmaChoice,
@@ -248,7 +249,9 @@ def test_get_participants():
 def test_factorial_participants():
     factorial_participants = get_participants(num_participant_samples=0)
     assert get_participants(0) == factorial_participants
-    assert len(factorial_participants) == 3888
+    assert len(factorial_participants) == len(
+        list(exhaustive(GROUP_PROMPT_CONDITIONS))
+    ) * len(list(exhaustive(PD_ATTRIBUTES)))
 
 
 def test_run_repeated_game(mocker, base_condition):