Rename load file functions

src/details.py

@@ -5,7 +5,7 @@ import pandas as pd
 from huggingface_hub import HfFileSystem
 
 import src.constants as constants
-from src.hub import load_details_file
+from src.hub import load_jsonlines_file
 
 
 def update_task_description_component(task):
@@ -53,7 +53,7 @@ async def load_details_dataframe(model_id, subtask):
     if not paths:
         return
     path = max(paths)
-    data = await load_details_file(path)
+    data = await load_jsonlines_file(path)
     df = pd.json_normalize(data)
     df = df.sort_values(by=["doc_id"])
     # df = df.rename_axis("Parameters", axis="columns")

src/hub.py

@@ -9,13 +9,13 @@ from huggingface_hub.utils import build_hf_headers
 client = httpx.AsyncClient(follow_redirects=True)
 
 
-async def load_file(path):
+async def load_json_file(path):
     url = to_url(path)
     r = await client.get(url)
     return r.json()
 
 
-async def load_details_file(path):
+async def load_jsonlines_file(path):
     url = to_url(path)
     r = await client.get(url, headers=build_hf_headers())
     f = io.StringIO(r.text)

src/results.py

@@ -5,7 +5,7 @@ import pandas as pd
 from huggingface_hub import HfFileSystem
 
 import src.constants as constants
-from src.hub import load_file
+from src.hub import load_json_file
 
 
 def fetch_result_paths():
@@ -32,7 +32,7 @@ async def load_results_dataframe(model_id, result_paths_per_model=None):
     if not model_id or not result_paths_per_model:
         return
     result_paths = result_paths_per_model[model_id]
-    results = await asyncio.gather(*[load_file(path) for path in result_paths])
+    results = await asyncio.gather(*[load_json_file(path) for path in result_paths])
     data = {"results": {}, "configs": {}}
     for result in results:
         data["results"].update(result["results"])