Add retrain_space_id to BaseConfig and update config validation

configs.py

@@ -30,6 +30,7 @@ class BaseConfig:
     hub_model_id: str = os.getenv("HUB_MODEL_ID", None)
     api_endpoint: str = os.getenv("API_ENDPOINT", None)
     space_repo_id: str = os.getenv("SPACE_REPO_ID", None)
+    retrain_space_id: str= os.getenv("RETRAIN_SPACE_ID", None)
 
     # Weights & Biases
     wandb_api_key: str = os.getenv("WANDB_API_KEY", None)
@@ -81,7 +82,8 @@ def get_config() -> BaseConfig:
             "hub_model_id",
             "api_endpoint",
             "space_repo_id",
-            "wandb_api_key"
+            "wandb_api_key",
+            "retrain_space_id"
         ]
     elif label == "urgency":
         config = UrgencyConfig()
@@ -91,7 +93,8 @@ def get_config() -> BaseConfig:
             "hub_model_id",
             "api_endpoint",
             "space_repo_id",
-            "wandb_api_key"
+            "wandb_api_key", 
+            "retrain_space_id"
         ]
     else:
         raise ValueError(f"Unsupported LABEL '{label}' in environment variables.")