update renamed variable file_set_id in FacemapModelTraining

datajoint · Oct 4, 2023 · a2df523 · a2df523
1 parent 3727884
commit a2df523
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/element_facemap/train_facial_model.py b/element_facemap/train_facial_model.py
@@ -293,12 +293,12 @@ def make(self, key):
 
         video_files = [find_full_path(fbe.get_facemap_root_data_dir(), fp).as_posix() 
                          for fp in (FacemapTrainFileSet.VideoFile & 
-                                    {'video_set_id': key['video_set_id']}).fetch("file_path")]
+                                    {'file_set_id': key['file_set_id']}).fetch("file_path")]
 
         # manually specified .h5 keypoints file 
         train_fileset = [find_full_path(fbe.get_facemap_root_data_dir(), fp).as_posix() 
                          for fp in (FacemapTrainFileSet.File & 
-                                    {'file_set_id': key['video_set_id']}).fetch("file_path")]
+                                    {'file_set_id': key['file_set_id']}).fetch("file_path")]
 
         keypoints_file_name = (FacemapModelTrainingTask & key).fetch1("keypoints_filename")