diff --git a/pyrelacs/repros/repros.py b/pyrelacs/repros/repros.py index 3067e09..fd197b8 100644 --- a/pyrelacs/repros/repros.py +++ b/pyrelacs/repros/repros.py @@ -44,10 +44,15 @@ class Repro: else: raise AttributeError(f"{file.name} has no {name} class") - def names_of_repros(self) -> Tuple[list, list]: + def names_of_repros(self, include_repros: list[str]) -> Tuple[list, list]: """ Searches for class names in the repro folder in all python files + Parameters + ---------- + include_repros : list[str] + List of repros to include in the pyrelacs instance + Returns ------- Tuple[list, list] @@ -73,4 +78,7 @@ class Repro: repro_names.extend(class_name) file_names.append(python_file) file.close() + + repro_names = [r for r in repro_names if r in include_repros] + file_names = [f for r, f in zip(repro_names, file_names) if r in include_repros] return repro_names, file_names