update to create output_dir during pose estimation and update task

datajoint · Jan 5, 2024 · f53852d · f53852d
1 parent 50f0d43
commit f53852d
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/element_deeplabcut/model.py b/element_deeplabcut/model.py
@@ -706,7 +706,14 @@ def make(self, key):
         task_mode, output_dir = (PoseEstimationTask & key).fetch1(
             "task_mode", "pose_estimation_output_dir"
         )
-
+        if not output_dir:
+            output_dir = PoseEstimationTask.infer_output_dir(
+                key, relative=True, mkdir=True
+            )
+            # update pose_estimation_output_dir
+            PoseEstimationTask.update1(
+                {**key, "pose_estimation_output_dir": output_dir.as_posix()}
+            )
         output_dir = find_full_path(get_dlc_root_data_dir(), output_dir)
 
         # Triger PoseEstimation