wuenlp_tools.utils.logging

View Source

 1import os
 2from typing import Optional
 3
 4from loguru import logger
 5
 6from wuenlp_tools import keys
 7
 8
 9def setup_wandb(project_name: str,
10                api_key: Optional[str] = None,
11                override: bool = False):
12    os.environ["WANDB_PROJECT"] = project_name
13
14    # Handle the api_key parameter
15    if api_key is not None:
16        os.environ["WANDB_API_KEY"] = api_key  # Always override if provided
17    elif override or "WANDB_API_KEY" not in os.environ:
18        os.environ["WANDB_API_KEY"] = keys.WANDB_API_KEY.data  # Override using keys if override is True
19
20    # Warning if either project_name or api_key is None after setting
21    # Checking the environment variables to determine if they are missing or set to 'None'
22    if os.environ.get("WANDB_PROJECT") is None or os.environ.get("WANDB_PROJECT") == "None":
23        logger.warning("WANDB_PROJECT is None after setup.")
24    if os.environ.get("WANDB_API_KEY") is None or os.environ.get("WANDB_API_KEY") == "None":
25        logger.warning("WANDB_API_KEY is None after setup.")

def setup_wandb( project_name: str, api_key: Optional[str] = None, override: bool = False): View Source

10def setup_wandb(project_name: str,
11                api_key: Optional[str] = None,
12                override: bool = False):
13    os.environ["WANDB_PROJECT"] = project_name
14
15    # Handle the api_key parameter
16    if api_key is not None:
17        os.environ["WANDB_API_KEY"] = api_key  # Always override if provided
18    elif override or "WANDB_API_KEY" not in os.environ:
19        os.environ["WANDB_API_KEY"] = keys.WANDB_API_KEY.data  # Override using keys if override is True
20
21    # Warning if either project_name or api_key is None after setting
22    # Checking the environment variables to determine if they are missing or set to 'None'
23    if os.environ.get("WANDB_PROJECT") is None or os.environ.get("WANDB_PROJECT") == "None":
24        logger.warning("WANDB_PROJECT is None after setup.")
25    if os.environ.get("WANDB_API_KEY") is None or os.environ.get("WANDB_API_KEY") == "None":
26        logger.warning("WANDB_API_KEY is None after setup.")