Skip to content

Commit

Permalink
cleanup outdated usages of PoseOutput
Browse files Browse the repository at this point in the history
  • Loading branch information
samuelbray32 committed Dec 20, 2024
1 parent 69335a8 commit fc46b37
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 8 deletions.
2 changes: 1 addition & 1 deletion src/spyglass/behavior/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def create_group(
group_name : str
Name of the group
keys : List[dict]
list of keys from PoseOutput to include in the group
list of keys from PositionOutput to include in the group
bodyparts : List[str], optional
body parts to include in the group, by default None includes all from every set
"""
Expand Down
10 changes: 3 additions & 7 deletions src/spyglass/behavior/moseq.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
import hashlib
import os
from pathlib import Path

Expand Down Expand Up @@ -322,9 +321,9 @@ def make(self, key):
num_iters = (MoseqSyllableSelection & key).fetch1("num_iters")

# load data and format for moseq
video_path = (PoseOutput & merge_key).fetch_video_path()
video_path = (PositionOutput & merge_key).fetch_video_path()
video_name = Path(video_path).stem + ".mp4"
bodyparts_df = (PoseOutput & merge_key).fetch_dataframe()
bodyparts_df = (PositionOutput & merge_key).fetch_pose_dataframe()

if bodyparts is None:
bodyparts = bodyparts_df.keys().get_level_values(0).unique().values
Expand All @@ -348,10 +347,7 @@ def make(self, key):
results_df.index = bodyparts_df.index

# save results into a nwbfile
nwb_file_name = PoseOutput.merge_get_parent(merge_key).fetch1(
"nwb_file_name"
)
nwb_file_name = PoseOutput.merge_get_parent(merge_key).fetch1(
nwb_file_name = PositionOutput.merge_get_parent(merge_key).fetch1(
"nwb_file_name"
)
analysis_file_name = AnalysisNwbfile().create(nwb_file_name)
Expand Down

0 comments on commit fc46b37

Please sign in to comment.