Tom Aarsen commited on
Commit
578a44d
1 Parent(s): 2e647d9

Avoid "0.41000000000000003" GB of memory usage

Browse files
Files changed (1) hide show
  1. refresh.py +2 -0
refresh.py CHANGED
@@ -715,6 +715,8 @@ def load_results(data_path: str) -> list | dict | pd.DataFrame | str | None:
715
  df = pd.read_json(data_path, orient="records", lines=True)
716
  if "index" in df.columns:
717
  df = df.set_index("index")
 
 
718
  return df
719
 
720
  else:
 
715
  df = pd.read_json(data_path, orient="records", lines=True)
716
  if "index" in df.columns:
717
  df = df.set_index("index")
718
+ if "Memory Usage (GB, fp32)" in df.columns:
719
+ df["Memory Usage (GB, fp32)"] = df["Memory Usage (GB, fp32)"].map(lambda value: round(value, 2) if isinstance(value, float) else value)
720
  return df
721
 
722
  else: