Dataframe → Series[Any]

SainsburyWellcomeCentre · Dec 20, 2023 · 9e11d54 · 9e11d54
1 parent abaec7d
commit 9e11d54
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/generate_sample_projects/pose_utils.py b/scripts/generate_sample_projects/pose_utils.py
@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Optional
+from typing import Any, Optional
 
 import pandas as pd
 from file_utils import check_file_io_safety
@@ -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[Any], 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.