cleanup setup_project call

LorenFrankLab · Jan 3, 2025 · bd29c76 · bd29c76
1 parent 5a33090
commit bd29c76
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/src/spyglass/behavior/v1/moseq.py b/src/spyglass/behavior/v1/moseq.py
@@ -128,19 +128,16 @@ def make(self, key):
             if os.path.lexists(destination):
                 os.remove(destination)  # remove if it's a broken symlink
             os.symlink(video, destination)
-        bodyparts = (PoseGroup & key).fetch1("bodyparts")
-        skeleton = model_params["skeleton"]
-        anterior_bodyparts = model_params.get("anterior_bodyparts", None)
-        posterior_bodyparts = model_params.get("posterior_bodyparts", None)
 
+        bodyparts = (PoseGroup & key).fetch1("bodyparts")
         kpms.setup_project(
             str(project_dir),
             video_dir=str(video_dir),
             bodyparts=bodyparts,
-            skeleton=skeleton,
+            skeleton=model_params["skeleton"],
             use_bodyparts=bodyparts,
-            anterior_bodyparts=anterior_bodyparts,
-            posterior_bodyparts=posterior_bodyparts,
+            anterior_bodyparts=model_params.get("anterior_bodyparts", None),
+            posterior_bodyparts=model_params.get("posterior_bodyparts", None),
         )
 
         config = self._config_func(project_dir)