improved docstrings

database/utils.py

@@ -276,6 +276,7 @@ def get_or_add_model_by_name(model: str, model_source: str = "hf"):
 
     Args:
         model (str): The path or identifier for the Hugging Face or other model.
+        model_source (str): Source of the model (as model arg in lm_eval or eval.py)
     """
     if model_source == "hf":
         git_commit_hash = HfApi().model_info(model).sha
@@ -384,7 +385,7 @@ def register_model_to_db(model_name: str, model_source: str) -> UUID:
 
     Args:
         model_name (str): The name or identifier for the model
-        model_source (str): Source of the model (e.g., 'openai-chat-completions', 'anthropic')
+        model_source (str): Source of the model (e.g., 'openai-chat-completions' or other model arg in lm_eval)
 
     Returns:
         UUID: The unique identifier assigned to the registered model

eval/eval_tracker.py

@@ -188,6 +188,7 @@ def get_or_create_model(
         Args:
             model_name: Name of the model
             model_id: Optional UUID of existing model
+            model_source: Source of the model (as model arg in lm_eval or eval.py)
 
         Returns:
             Tuple of (model_id, dataset_id)
@@ -347,7 +348,7 @@ def update_evalresults_db(
         Args:
             eval_log_dict: Dictionary containing evaluation logs and results
             model_id: Optional UUID of the model
-            model_source: Source of the model (similar to the model arg in eval.py)
+            model_source: Source of the model (similar to the model arg in lm_eval or eval.py)
             model_name: Optional name of the model
             creation_location: Location where evaluation was run
             created_by: Username who ran the evaluation