jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix linting error

Browse Source
This commit is contained in:
DseidLi 2024-02-28 16:38:31 +08:00
parent b6524f3ebf
commit 1c7375681d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
opencompass/summarizers/needlebench.py
View File

@ -49,7 +49,7 @@ def read_after_specific_line_except_last(file_name, keyword, offset):
def create_model_dataframe(nested_dict, model_name, dataset_abbr, parallel=False):
if model_name not in nested_dict:
print(f'Model {model_name} not found in the provided data.')
return pd.DataFrame()
return pd.DataFrame()
model_data = nested_dict[model_name]
data = []
@ -267,7 +267,7 @@ def merge_dataframes(model_name, dataset_abbrs, parsed_data):
from functools import reduce
merged_df = reduce(lambda left, right: pd.merge(left, right, on='dataset', how='outer'), dfs)
if merged_df.isnull().any().any():
print('Warning: Some rows were filtered out due to NaN values. This is often due to mismatched row counts among DataFrames.')
merged_df = merged_df.dropna()