Dedupe types, share defaults

Author: farrelmahaztra

src/inspect_evals/musr/musr.py

@@ -29,14 +29,19 @@
     TEAM_ALLOCATION_HINT,
 )
 
+DomainType = Literal["murder_mysteries", "object_placements", "team_allocation"]
+PromptTechniqueType = Literal["regular", "cot", "cot+"]
+
+DEFAULT_DOMAIN: DomainType = "murder_mysteries"
+DEFAULT_PROMPT_TECHNIQUE: PromptTechniqueType = "regular"
+DEFAULT_EXAMPLE_COUNT = 0
+
 
 @task
 def musr(
-    domain: Literal[
-        "murder_mysteries", "object_placements", "team_allocation"
-    ] = "murder_mysteries",
-    prompt_technique: Literal["regular", "cot", "cot+"] = "regular",
-    example_count: int = 0,
+    domain: DomainType = DEFAULT_DOMAIN,
+    prompt_technique: PromptTechniqueType = DEFAULT_PROMPT_TECHNIQUE,
+    example_count: int = DEFAULT_EXAMPLE_COUNT,
 ) -> Task:
     """Inspect task implementing the MuSR benchmark.
 
@@ -68,9 +73,9 @@ def musr(
 
 
 def get_domain_prompt(
-    domain: Literal["murder_mysteries", "object_placements", "team_allocation"],
-    prompt_technique: Literal["regular", "cot", "cot+"],
-    example_count: int,
+    domain: DomainType = DEFAULT_DOMAIN,
+    prompt_technique: PromptTechniqueType = DEFAULT_PROMPT_TECHNIQUE,
+    example_count: int = DEFAULT_EXAMPLE_COUNT,
 ) -> str:
     domain_info = {
         "murder_mysteries": {