fix

mlcommons · Oct 7, 2023 · 2598d39 · 2598d39
1 parent 557bf0d
commit 2598d39
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/algorithmic_efficiency/logger_utils.py b/algorithmic_efficiency/logger_utils.py
@@ -262,7 +262,7 @@ def _get_workload_properties(workload: spec.Workload) -> dict:
   return workload_properties
 
 
-def get_meta_data(workload: spec.Workload) -> dict:
+def get_meta_data(workload: spec.Workload, rng_seed: int = None) -> dict:
   meta_data = {}
   workload_properties = _get_workload_properties(workload)
   meta_data.update(workload_properties)
@@ -272,14 +272,8 @@ def get_meta_data(workload: spec.Workload) -> dict:
   meta_data.update(system_software_info)
   system_hardware_info = _get_system_hardware_info()
   meta_data.update(system_hardware_info)
-  return meta_data
-
-
-def get_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})
-  # with open(meta_file_name, 'w') as f:
-  #   f.write(json.dumps(meta_data, indent=2))
+  if rng_seed:
+    meta_data.update({'rng_seed': rng_seed}) 
   return meta_data