jasonshaoshun commited on
Commit
a0adce1
·
1 Parent(s): c757005
Files changed (1) hide show
  1. src/leaderboard/read_evals.py +5 -1
src/leaderboard/read_evals.py CHANGED
@@ -404,7 +404,11 @@ def get_raw_eval_results_mib_causalgraph(results_path: str) -> Tuple[pd.DataFram
404
  # Check if this is the consolidated format by examining a sample key
405
  sample_key = next(iter(data), None)
406
  if sample_key and isinstance(sample_key, str) and '(' in sample_key and ')' in sample_key:
407
- raw_data = data
 
 
 
 
408
  print(f"Found consolidated data file: {json_file}")
409
 
410
  except Exception as e:
 
404
  # Check if this is the consolidated format by examining a sample key
405
  sample_key = next(iter(data), None)
406
  if sample_key and isinstance(sample_key, str) and '(' in sample_key and ')' in sample_key:
407
+ if raw_data is None:
408
+ raw_data = data
409
+ else:
410
+ # If we already have data, we can merge it
411
+ raw_data.update(data)
412
  print(f"Found consolidated data file: {json_file}")
413
 
414
  except Exception as e: