diff --git a/iblrig/base_tasks.py b/iblrig/base_tasks.py index 2c16d2598..7a60cdfaa 100644 --- a/iblrig/base_tasks.py +++ b/iblrig/base_tasks.py @@ -1591,6 +1591,8 @@ def _run(self): break if self.stopped: break + # Create the transfer flag file when the session finishes. + self.paths.SESSION_FOLDER.joinpath('transfer_me.flag').touch() class SpontaneousBpodSession(SpontaneousSession, BpodMixin): diff --git a/iblrig_tasks/_iblrig_tasks_passiveChoiceWorld/task.py b/iblrig_tasks/_iblrig_tasks_passiveChoiceWorld/task.py index 775f07b7a..e62cdad33 100644 --- a/iblrig_tasks/_iblrig_tasks_passiveChoiceWorld/task.py +++ b/iblrig_tasks/_iblrig_tasks_passiveChoiceWorld/task.py @@ -125,6 +125,8 @@ def _run(self): self.bonsai_visual_udp_client.send_message(r'/re', byte_hide_stim) if self.stopped: break + # Create the transfer flag when the task finishes. + self.paths.SESSION_FOLDER.joinpath('transfer_me.flag').touch() if __name__ == '__main__': # pragma: no cover