Skip to content

Commit bf5cf9c

Browse files
committed
Add training workflow for DR
1 parent 9a82367 commit bf5cf9c

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

src/np_workflows/experiments/dynamic_routing/main.py

+12
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,8 @@ def new_experiment(
5656
experiment = Hab(mouse, user)
5757
elif 'OPTO' in workflow.name:
5858
experiment = Opto(mouse, user)
59+
elif 'TRAINING' in workflow.name:
60+
experiment = Training(mouse, user)
5961
else:
6062
raise ValueError(f"Unknown {workflow = }. Create an experiment with e.g.\n\n\texperiment = Ephys(mouse, user)\nexperiment.session.npexp_path.mkdir()")
6163
experiment.workflow = workflow
@@ -102,4 +104,14 @@ def __init__(self, *args, **kwargs):
102104
ScriptCamstim,
103105
OpenEphys,
104106
)
107+
super().__init__(*args, **kwargs)
108+
109+
class Training(DRTask):
110+
def __init__(self, *args, **kwargs):
111+
self.services = (
112+
MouseDirector,
113+
Sync,
114+
VideoMVR,
115+
ScriptCamstim,
116+
)
105117
super().__init__(*args, **kwargs)

src/np_workflows/shared/base_experiments.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -475,7 +475,7 @@ class Workflow(enum.Enum):
475475
HAB = "EPHYS minus probes"
476476
EPHYS = "opto in task optional"
477477
OPTO = "opto in task, no ephys"
478-
478+
TRAINING = "task only, with sync + video"
479479
workflow: Workflow
480480

481481
@property

0 commit comments

Comments
 (0)