Fix the relative vs. absolute path issues

There are some small issues when loading the datasets / examples from the config, due to the recent change from "backend/omnisealbench/data" to "backend/data". We fix this with the better handling of the default value of 'OMNISEAL_LEADERBOARD_DATA'

backend/config.py

@@ -2,6 +2,7 @@
 # IMPORTANT: When running from docker more setup is required (e.g. on Huggingface)
 import os
 from collections import defaultdict
+from pathlib import Path
 
 ABS_DATASET_DOMAIN = "https://dl.fbaipublicfiles.com"
 
@@ -139,7 +140,8 @@ DATASET_CONFIGS = {
 
 def get_user_dataset():
     datasets = defaultdict(list)
-    user_data_dir = os.getenv("OMNISEAL_LEADERBOARD_DATA", "./data")
+    default_local_data_dir = str(Path(__file__).parent.joinpath("data"))
+    user_data_dir = os.getenv("OMNISEAL_LEADERBOARD_DATA", default_local_data_dir)
     if user_data_dir:
         for user_data in os.listdir(user_data_dir):
             if not os.path.isdir(os.path.join(user_data_dir, user_data)):
@@ -171,7 +173,7 @@ def get_datasets():
 def get_example_config(type, dataset_name):
     """Get example configuration for a specific dataset."""
     if not dataset_name:
-        raise ValueError(f"Dataset name is required")
+        raise ValueError("Dataset name is required")
 
     # Check if it's a valid dataset for this type
     all_datasets = get_datasets()
@@ -199,7 +201,7 @@ def get_example_config(type, dataset_name):
             examples_config = {
                 "dataset_name": dataset_base_name,
                 "path": dataset_config["path"],
-                "db_key": _get_db_key_for_dataset(dataset_base_name, type),
+                "db_key": _get_db_key_for_dataset(dataset_base_name),
             }
         else:
             raise ValueError(f"Dataset {dataset_name} not found in configurations")
@@ -207,7 +209,7 @@ def get_example_config(type, dataset_name):
     return examples_config
 
 
-def _get_db_key_for_dataset(dataset_base_name, type):
+def _get_db_key_for_dataset(dataset_base_name):
     """Helper function to determine the database key for a dataset"""
     # Map of dataset names to their db keys
     db_key_mapping = {
@@ -230,10 +232,12 @@ def get_dataset_config(dataset_name):
     else:
         modality = dataset_name.split("/")[-1]
         user_dataset = get_user_dataset()
+        default_local_data_dir = str(Path(__file__).parent.joinpath("data"))
+        user_data_dir = os.getenv("OMNISEAL_LEADERBOARD_DATA", default_local_data_dir)
         if dataset_name in user_dataset.get(modality, []):
             cfg = {
                 "type": modality,
-                "path": os.getenv("OMNISEAL_LEADERBOARD_DATA", "./data"),
+                "path": user_data_dir,
             }
             extra_cfg = MODALITY_CONFIG_CONSTANTS.get(cfg["type"], {})
             cfg.update(extra_cfg)