Merge pull request AIRI-Institute#130 from AIRI-Institute/plot

small changes
bigscience-workshop · Aug 28, 2023 · aa7af38 · aa7af38
2 parents 2418575 + ee263a8
commit aa7af38
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/probing/basic_plot.py b/probing/basic_plot.py
@@ -39,10 +39,10 @@ def __init__(
         self.value_field = value_field
 
     @staticmethod
-    def get_logs(paths: List[Path]) -> List[Path]:
+    def get_logs(paths: List[Path], filename: str = "log.json") -> List[Path]:
         logs_path = []
         for path in paths:
-            internal_log_paths = path.glob("**/*/log.json")
+            internal_log_paths = path.glob(f"**/*/{filename}")
             for log_path in internal_log_paths:
                 if log_path not in logs_path:
                     logs_path.append(log_path)