Skip to content

Commit

Permalink
Dataframe → Series[Any]
Browse files Browse the repository at this point in the history
  • Loading branch information
samcunliffe committed Dec 20, 2023
1 parent abaec7d commit 9e11d54
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions scripts/generate_sample_projects/pose_utils.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down Expand Up @@ -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.
Expand Down

0 comments on commit 9e11d54

Please sign in to comment.