diff --git a/scripts/generate_sample_projects/pose_utils.py b/scripts/generate_sample_projects/pose_utils.py index 239caf6..ff2eabd 100644 --- a/scripts/generate_sample_projects/pose_utils.py +++ b/scripts/generate_sample_projects/pose_utils.py @@ -33,7 +33,7 @@ def load_poses_from_dlc(file_path: Path) -> Optional[pd.DataFrame]: return df -def save_poses_to_dlc(df: pd.DataFrame, file_path: Path): +def save_poses_to_dlc(df: pd.Series, file_path: Path): """Save pose estimation results to a DeepLabCut (DLC) .h5 file. Also saves the poses to a .csv file with the same name.