wip

app.py

@@ -29,15 +29,18 @@ from src.display.css_html_js import custom_css
 #     Precision
 # )
 from src.envs import API, REPO_ID, TOKEN, CODE_PROBLEMS_REPO, SUBMISSIONS_REPO, RESULTS_REPO
+from src.logger import get_logger
 # from src.populate import get_evaluation_queue_df, get_leaderboard_df
 from src.submission.submit import add_new_solutions
 
+logger = get_logger(__name__)
 
 def restart_space():
     API.restart_space(repo_id=REPO_ID)
 
 lbdb = F1Data(cp_ds_name=CODE_PROBLEMS_REPO, sub_ds_name=SUBMISSIONS_REPO, res_ds_name=RESULTS_REPO)
 
+logger.info("Initialized LBDB")
 
 # (
 #     finished_eval_queue_df,
@@ -87,9 +90,11 @@ with demo:
         #     leaderboard = init_leaderboard(LEADERBOARD_DF)
 
         with gr.TabItem("📝 About", elem_id="llm-benchmark-tab-table", id=2):
+            logger.info("Tab about")
             gr.Markdown(LLM_BENCHMARKS_TEXT, elem_classes="markdown-text")
 
         with gr.TabItem("🚀 Submit here! ", elem_id="llm-benchmark-tab-table", id=3):
+            logger.info("Tab submission")
             with gr.Column():
                 with gr.Row():
                     gr.Markdown(EVALUATION_QUEUE_TEXT, elem_classes="markdown-text")

src/datamodel/data.py

@@ -1,4 +1,5 @@
 import functools
+import time
 
 from datasets import load_dataset
 
@@ -20,5 +21,8 @@ class F1Data:
 
     def initialize(self):
         logger.info("Initialize F1Data TOMEN='%s'", TOKEN)
+        start_time = time.monotonic()
         cp_ds = load_dataset(self.cp_dataset_name, split="hard", token=TOKEN)
+        logger.info("Loaded code-problems dataset from %s in %f sec", self.cp_dataset_name, time.monotonic() - start_time)
         self.code_problems: dict[str, str] = {r["formula_name"]: r["code_problem"]["problem_description"] for r in cp_ds}
+        logger.info("Code problems info: %s", self.code_problems)