From 64cd1b00ad9220cb576760056665bcb2807d870b Mon Sep 17 00:00:00 2001
From: wendtalexander <wendtalexander@protonmail.com>
Date: Fri, 18 Oct 2024 16:30:46 +0200
Subject: [PATCH] [repros] updating names of repros
---
pyrelacs/repros/repros.py | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
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