Skip to content

Commit

Permalink
Add training workflow for DR
Browse files Browse the repository at this point in the history
  • Loading branch information
bjhardcastle committed Aug 12, 2024
1 parent 9a82367 commit bf5cf9c
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 1 deletion.
12 changes: 12 additions & 0 deletions src/np_workflows/experiments/dynamic_routing/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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)
2 changes: 1 addition & 1 deletion src/np_workflows/shared/base_experiments.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit bf5cf9c

Please sign in to comment.