Add register function for eval experiments.

PiperOrigin-RevId: 533549209

putting_dune/experiments/registry.py

@@ -34,7 +34,6 @@
 from putting_dune.experiments import experiments
 
 
-
 # -------------------- AGENTS --------------------
 
 
@@ -369,6 +368,16 @@ def _get_human_prior_rates_config() -> experiments.SimulatorConfig:
 )
 
 
+def register_eval_experiment(
+    name: str, eval_experiment: experiments.EvalExperiment
+):
+  global _EVAL_EXPERIMENTS
+  if name not in _EVAL_EXPERIMENTS:
+    _EVAL_EXPERIMENTS = frozendict.frozendict(
+        {name: eval_experiment, **_EVAL_EXPERIMENTS}
+    )
+
+
 def create_microscope_experiment(name: str) -> experiments.MicroscopeExperiment:
   if name not in _MICROSCOPE_EXPERIMENTS:
     raise ValueError(f'Unknown microscope experiment {name}.')