wip

src/datamodel/data.py

@@ -2,6 +2,11 @@ import functools
 
 from datasets import load_dataset
 
+from src.envs import TOKEN
+from src.logger import get_logger
+
+logger = get_logger(__name__)
+
 class F1Data:
     def __init__(self, cp_ds_name: str, sub_ds_name: str, res_ds_name: str):
         self.cp_dataset_name = cp_ds_name
@@ -14,8 +19,6 @@ class F1Data:
         return set(self.code_problems.keys())
 
     def initialize(self):
+        logger.info("Initialize F1Data TOMEN='%s'", TOKEN)
         cp_ds = load_dataset(self.cp_dataset_name, split="hard")
         self.code_problems: dict[str, str] = {r["formula_name"]: r["code_problem"]["problem_description"] for r in cp_ds}
-
-    def add_submission(self, submitter: str, submission_path: str):
-        pass

src/logger.py

@@ -0,0 +1,11 @@
+import logging
+import sys
+
+def get_logger(filename: str, level=logging.INFO) -> logging.Logger:
+    new_logger = logging.getLogger(filename)
+    fmt = logging.Formatter(fmt="%(asctime)s - %(name)s - %(levelname)s - %(funcName)s:%(lineno)d - %(message)s")
+    handler = logging.StreamHandler(sys.stderr)
+    handler.setFormatter(fmt)
+    new_logger.addHandler(handler)
+    new_logger.setLevel(level)
+    return new_logger