add Literal type

Author: xzrderek

eval_protocol/pytest/evaluation_test.py

@@ -72,12 +72,13 @@
 from ..common_utils import load_jsonl
 
 from pytest import StashKey
+from typing_extensions import Literal
 
 
 EXPERIMENT_LINKS_STASH_KEY = StashKey[list]()
 
 
-def _store_experiment_link(experiment_id: str, job_link: str, status: str):
+def _store_experiment_link(experiment_id: str, job_link: str, status: Literal["success", "failure"]):
     """Store experiment link in pytest session stash."""
     try:
         import sys