From 554d87fab261304da0a4db2efe9a24863bf96491 Mon Sep 17 00:00:00 2001
From: Sam Cunliffe <s.cunliffe@ucl.ac.uk>
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..ec7e0d3 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.DataFrame | 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.