From 6c2d48802c9f30ade649c96f94e98df46a4456e1 Mon Sep 17 00:00:00 2001 From: Ben Hardcastle Date: Tue, 7 Jan 2025 15:30:35 -0800 Subject: [PATCH] Fix copying for behavior.pkl files --- src/np_workflows/shared/base_experiments.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/np_workflows/shared/base_experiments.py b/src/np_workflows/shared/base_experiments.py index 0674e07..641c82b 100644 --- a/src/np_workflows/shared/base_experiments.py +++ b/src/np_workflows/shared/base_experiments.py @@ -411,7 +411,9 @@ def copy_data_files(self) -> None: for _ in ('opto', 'main', 'mapping', 'behavior'): if _ in file.name: renamed = f'{self.session.folder}.{"stim" if _ == "main" else _}.pkl' - elif self.contains_uuid(file.name): + break + else: + if self.contains_uuid(file.name): renamed = f'{self.session.folder}.behavior.pkl' elif file.suffix in ('.json', '.mp4') and (cam_label := re.match('Behavior|Eye|Face',file.name)): renamed = f'{self.session.folder}.{cam_label.group().lower()}{file.suffix}'