feat(LabeledVideo): add labeled_video_file attribute as filepath

kushalbakshi · Aug 12, 2024 · 8c5a5e9 · 8c5a5e9
1 parent 845528c
commit 8c5a5e9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/element_deeplabcut/model.py b/element_deeplabcut/model.py
@@ -931,6 +931,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
@@ -1008,6 +1009,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/setup.py b/setup.py
@@ -48,6 +48,6 @@
         "tests": ["pytest", "pytest-cov", "shutils"],
         "dlc-pytorch": [
             "deeplabcut @ git+https://github.com/DeepLabCut/DeepLabCut.git@pytorch_dlc"
-        ]
+        ],
     },
 )