import pandas as pd # type: ignore[import] | |
def get_results_for_task_stub(task: str) -> pd.DataFrame: | |
stub_df = pd.DataFrame( | |
[ | |
{ | |
"Model Name": "GPT-4", | |
"Availability": "Proprietary", | |
"Context Size": "16k", | |
"BLEU": "X", | |
"ROUGE": "X", | |
"ChrF": "X", | |
"BERTScore": "X", | |
"BERTScore (Normalized)": "X", | |
"Submitted By": "π Long Code Arena Team", | |
}, | |
{ | |
"Model Name": "CodeLlama-7b (instruct)", | |
"Availability": "Llama 2 license", | |
"Context Size": "16k", | |
"BLEU": "X", | |
"ROUGE": "X", | |
"ChrF": "X", | |
"BERTScore": "X", | |
"BERTScore (Normalized)": "X", | |
"Submitted By": "π Long Code Arena Team", | |
}, | |
] | |
) | |
return stub_df | |