Merge pull request #32 from sidhulyalkar/codespace

Minor update related to name changes
datajoint · Jan 23, 2024 · 4c407b6 · 4c407b6
2 parents 0004f4b + 7ba9dcb
commit 4c407b6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/element_facemap/facemap_inference.py b/element_facemap/facemap_inference.py
@@ -286,7 +286,7 @@ def generate(
             ),
         )
 
-    insert_pose_estimation_task = generate
+    insert_facemap_inference_task = generate
 
 
 @schema
@@ -338,7 +338,7 @@ def make(self, key):
             output_dir = FacemapInferenceTask.infer_output_dir(
                 key, relative=True, mkdir=True
             )
-            # update pose_estimation_output_dir
+            # update facemap_inference_output_dir
             FacemapInferenceTask.update1(
                 {**key, "facemap_inference_output_dir": output_dir.as_posix()}
             )
@@ -356,7 +356,7 @@ def make(self, key):
         facemap_result_path = output_dir / f"{vid_name}_FacemapPose.h5"
         full_metadata_path = output_dir / f"{vid_name}_FacemapPose_metadata.pkl"
 
-        # Trigger Facemap Pose Estimation Inference
+        # Load or Trigger Facemap Pose Estimation Inference
         if (
             facemap_result_path.exists() & full_metadata_path.exists()
         ) or task_mode == "load":  # Load results and do not rerun processing