From ee263a8534d5504c2b4001da24b8b086ad7c33d5 Mon Sep 17 00:00:00 2001 From: Vitaly Protasov Date: Mon, 28 Aug 2023 18:46:23 +0300 Subject: [PATCH] small changes --- probing/basic_plot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/probing/basic_plot.py b/probing/basic_plot.py index d2923c6..583dec1 100644 --- 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)