diff --git a/algorithmic_efficiency/logger_utils.py b/algorithmic_efficiency/logger_utils.py index 559859515..18652dcaa 100644 --- a/algorithmic_efficiency/logger_utils.py +++ b/algorithmic_efficiency/logger_utils.py @@ -275,8 +275,8 @@ def get_meta_data(workload: spec.Workload) -> dict: return meta_data -def save_meta_data(workload: spec.Workload, - rng_seed: int, +def save_meta_data(workload: spec.Workload, + rng_seed: int, meta_file_name: str): meta_data = get_meta_data(workload) meta_data.update({'rng_seed': rng_seed})