[repros] updating names of repros

This commit is contained in:
wendtalexander 2024-10-18 16:30:46 +02:00
parent f703687ed7
commit 64cd1b00ad

View File

@ -44,10 +44,15 @@ class Repro:
else: else:
raise AttributeError(f"{file.name} has no {name} class") 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 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 Returns
------- -------
Tuple[list, list] Tuple[list, list]
@ -73,4 +78,7 @@ class Repro:
repro_names.extend(class_name) repro_names.extend(class_name)
file_names.append(python_file) file_names.append(python_file)
file.close() 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 return repro_names, file_names