Dataframe → Series[Any]

SainsburyWellcomeCentre · Dec 20, 2023 · 850ddd3 · 850ddd3
1 parent 4993bbc
commit 850ddd3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.