From 5fa0a7e02641abacf41b128f2559d911561d4373 Mon Sep 17 00:00:00 2001 From: Thinh Nguyen Date: Fri, 19 Apr 2024 17:15:10 -0500 Subject: [PATCH 1/2] update(tracking): raise error if no SLEAP data found --- aeon/dj_pipeline/tracking.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/aeon/dj_pipeline/tracking.py b/aeon/dj_pipeline/tracking.py index 7e140252..f46d9ec6 100644 --- a/aeon/dj_pipeline/tracking.py +++ b/aeon/dj_pipeline/tracking.py @@ -172,8 +172,7 @@ def make(self, key): ) if not len(pose_data): - self.insert1(key) - return + raise ValueError(f"No SLEAP data found for {device_name}") # Find the config file for the SLEAP model for data_dir in data_dirs: From a1a2f44ddd5ae84111e2315bc76716a090f5042c Mon Sep 17 00:00:00 2001 From: Thinh Nguyen Date: Fri, 19 Apr 2024 17:22:00 -0500 Subject: [PATCH 2/2] chore: update error message --- aeon/dj_pipeline/tracking.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aeon/dj_pipeline/tracking.py b/aeon/dj_pipeline/tracking.py index f46d9ec6..e7c788fc 100644 --- a/aeon/dj_pipeline/tracking.py +++ b/aeon/dj_pipeline/tracking.py @@ -172,7 +172,7 @@ def make(self, key): ) if not len(pose_data): - raise ValueError(f"No SLEAP data found for {device_name}") + raise ValueError(f"No SLEAP data found for {key['experiment_name']} - {device_name}") # Find the config file for the SLEAP model for data_dir in data_dirs: