From 76bf0dc18709bb338973345c065eab86dca447da Mon Sep 17 00:00:00 2001
From: Sidharth Hulyalkar <sidharth.hulyalkar@datajoint.com>
Date: Fri, 5 Jan 2024 13:15:21 -0800
Subject: [PATCH] create output directory and update task in pose estimation
 step

---
 element_facemap/facemap_inference.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/element_facemap/facemap_inference.py b/element_facemap/facemap_inference.py
index c107c44..8eb8d3d 100644
--- a/element_facemap/facemap_inference.py
+++ b/element_facemap/facemap_inference.py
@@ -336,6 +336,15 @@ def make(self, key):
             "task_mode", "pose_estimation_output_dir"
         )
 
+        if not output_dir:
+            output_dir = FacemapPoseEstimationTask.infer_output_dir(
+                key, relative=True, mkdir=True
+            )
+            # update pose_estimation_output_dir
+            FacemapPoseEstimationTask.update1(
+                {**key, "pose_estimation_output_dir": output_dir.as_posix()}
+            )
+
         output_dir = find_full_path(fbe.get_facemap_root_data_dir(), output_dir)
         video_files = (FacemapPoseEstimationTask * fbe.VideoRecording.File & key).fetch(
             "file_path"