diff --git a/element_deeplabcut/model.py b/element_deeplabcut/model.py index 61341e5..2c02843 100644 --- a/element_deeplabcut/model.py +++ b/element_deeplabcut/model.py @@ -887,6 +887,7 @@ class File(dj.Part): -> VideoRecording.File --- labeled_video_path: varchar(255) # relative path to labeled video + labeled_video_file=null: filepath@dlc-processed """ @property @@ -964,6 +965,7 @@ def make(self, key): "labeled_video_path": labeled_video_path.relative_to( get_dlc_processed_data_dir() ).as_posix(), + "labeled_video_file": labeled_video_path.as_posix(), } ) diff --git a/tests/tutorial_pipeline.py b/tests/tutorial_pipeline.py index dd502d3..e3a8ae7 100644 --- a/tests/tutorial_pipeline.py +++ b/tests/tutorial_pipeline.py @@ -10,8 +10,12 @@ from element_lab.lab import Source, Lab, Protocol, User, Project -if "custom" not in dj.config: - dj.config["custom"] = {} +os.environ["DJ_SUPPORT_FILEPATH_MANAGEMENT"] = "TRUE" +dj.config["filepath_checksum_size_limit"] = 10 * 1024 * 1024 # 10 MB + +for key in ("custom", "stores"): + if key not in dj.config: + dj.config[key] = {} # overwrite dj.config['custom'] values with environment variables if available @@ -95,7 +99,12 @@ class Device(dj.Lookup): # Activate DeepLabCut schema ----------------------------------- - +# Configure external storage ------------- +dj.config["stores"]["dlc-processed"] = dict( + protocol="file", + location=get_dlc_processed_data_dir(), + stage=get_dlc_processed_data_dir(), +) train.activate(db_prefix + "train", linking_module=__name__) model.activate(db_prefix + "model", linking_module=__name__)