diff --git a/src/np_workflows/experiments/dynamic_routing/main.py b/src/np_workflows/experiments/dynamic_routing/main.py index 464f403..4961ee1 100644 --- a/src/np_workflows/experiments/dynamic_routing/main.py +++ b/src/np_workflows/experiments/dynamic_routing/main.py @@ -56,6 +56,8 @@ def new_experiment( experiment = Hab(mouse, user) elif 'OPTO' in workflow.name: experiment = Opto(mouse, user) + elif 'TRAINING' in workflow.name: + experiment = Training(mouse, user) else: raise ValueError(f"Unknown {workflow = }. Create an experiment with e.g.\n\n\texperiment = Ephys(mouse, user)\nexperiment.session.npexp_path.mkdir()") experiment.workflow = workflow @@ -102,4 +104,14 @@ def __init__(self, *args, **kwargs): ScriptCamstim, OpenEphys, ) + super().__init__(*args, **kwargs) + +class Training(DRTask): + def __init__(self, *args, **kwargs): + self.services = ( + MouseDirector, + Sync, + VideoMVR, + ScriptCamstim, + ) super().__init__(*args, **kwargs) \ No newline at end of file diff --git a/src/np_workflows/shared/base_experiments.py b/src/np_workflows/shared/base_experiments.py index ecb1ba0..98b0424 100644 --- a/src/np_workflows/shared/base_experiments.py +++ b/src/np_workflows/shared/base_experiments.py @@ -475,7 +475,7 @@ class Workflow(enum.Enum): HAB = "EPHYS minus probes" EPHYS = "opto in task optional" OPTO = "opto in task, no ephys" - + TRAINING = "task only, with sync + video" workflow: Workflow @property