From bf51a8f29886e516029da81be844db34fb63b30d Mon Sep 17 00:00:00 2001 From: Sam Bray Date: Thu, 16 Nov 2023 11:28:22 -0800 Subject: [PATCH] Add video directory path when getting recorded video --- src/spyglass/common/common_position.py | 4 ++-- src/spyglass/settings.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/spyglass/common/common_position.py b/src/spyglass/common/common_position.py index 1fb6f586f..4f03023af 100644 --- a/src/spyglass/common/common_position.py +++ b/src/spyglass/common/common_position.py @@ -23,7 +23,7 @@ plot_track_graph, ) -from ..settings import raw_dir +from ..settings import raw_dir, video_dir from ..utils.dj_helper_fn import fetch_nwb from .common_behav import RawPosition, VideoFile from .common_interval import IntervalList # noqa F401 @@ -935,7 +935,7 @@ def make(self, key): print("Making video...") self.make_video( - video_filename, + f"{video_dir}/{video_filename}", centroids, head_position_mean, head_orientation_mean, diff --git a/src/spyglass/settings.py b/src/spyglass/settings.py index 7f1ff9572..6a6dda2c8 100644 --- a/src/spyglass/settings.py +++ b/src/spyglass/settings.py @@ -442,3 +442,4 @@ def debug_mode(self) -> bool: sorting_dir = sg_config.sorting_dir waveform_dir = sg_config.waveform_dir debug_mode = sg_config.debug_mode +video_dir = sg_config.video_dir