From 850ddd33bebb97a364c067b9716a1afe3941f70d Mon Sep 17 00:00:00 2001 From: Sam Cunliffe Date: Wed, 20 Dec 2023 20:34:55 +0000 Subject: [PATCH] =?UTF-8?q?Dataframe=20=E2=86=92=20Series[Any]?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- scripts/generate_sample_projects/pose_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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.