diff --git a/sleap_roots/series.py b/sleap_roots/series.py index 76acb1f..5b98228 100644 --- a/sleap_roots/series.py +++ b/sleap_roots/series.py @@ -366,7 +366,7 @@ def get_crown_points(self, frame_idx: int) -> np.ndarray: return crown_pts -def find_all_series(data_folders: Union[str, List[str]]) -> List[str]: +def find_all_h5_series(data_folders: Union[str, List[str]]) -> List[str]: """Find all .h5 series from a list of folders. Args: @@ -384,6 +384,10 @@ def find_all_series(data_folders: Union[str, List[str]]) -> List[str]: return h5_series +def find_all_slp_series(data_folders: Union[str, List[str]]) -> List[str]: + """Find all .slp series from a list of folders.""" + + def imgfig( size: Union[float, Tuple] = 6, dpi: int = 72, scale: float = 1.0 ) -> matplotlib.figure.Figure: