From 830a604bebb627f27165f0b40c908dfa18744f5f Mon Sep 17 00:00:00 2001 From: Elizabeth Berrigan Date: Tue, 7 May 2024 18:01:09 -0700 Subject: [PATCH] find all slp series --- sleap_roots/series.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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: