fix: remove setup_metta_environment aws and wandb checks (#1129)

### TL;DR

Remove the AWS and Wandb credential checking functions from the
codebase.


[Asana
Task](https://app.asana.com/1/1209016784099267/project/1210348820405981/task/1210632382857191)

[Asana
Task](https://app.asana.com/1/1209016784099267/project/1210348820405981/task/1210632060208346)

---------

Co-authored-by: Robb Walters <robb@2amlogic.com>

Author: daveey

metta/util/config.py

@@ -1,12 +1,8 @@
 from __future__ import annotations
 
-import os
 from typing import Any, Dict, Optional, TypeVar, cast
 
-import boto3
 import hydra
-import wandb
-from botocore.exceptions import ClientError, NoCredentialsError
 from omegaconf import DictConfig, ListConfig, OmegaConf
 from pydantic import BaseModel
 
@@ -91,46 +87,3 @@ def config_from_path(config_path: str, overrides: Optional[DictConfig | ListConf
         cfg = OmegaConf.merge(cfg, overrides)
         OmegaConf.set_struct(cfg, True)
     return cast(DictConfig, cfg)
-
-
-def check_aws_credentials() -> bool:
-    """Check if valid AWS credentials are available from any source."""
-    if "AWS_ACCESS_KEY_ID" in os.environ and "AWS_SECRET_ACCESS_KEY" in os.environ:
-        # This check is primarily for github actions.
-        return True
-    try:
-        sts = boto3.client("sts")
-        sts.get_caller_identity()
-        return True
-    except (NoCredentialsError, ClientError):
-        return False
-
-
-def check_wandb_credentials() -> bool:
-    """Check if valid W&B credentials are available."""
-    if "WANDB_API_KEY" in os.environ:
-        # This check is primarily for github actions.
-        return True
-    try:
-        return wandb.login(anonymous="never", timeout=10)
-    except Exception:
-        return False
-
-
-def setup_metta_environment(cfg: ListConfig | DictConfig, require_aws: bool = True, require_wandb: bool = True):
-    if require_aws:
-        # Check that AWS is good to go.
-        if not check_aws_credentials():
-            print("AWS is not configured, please install:")
-            print("brew install awscli")
-            print("and run:")
-            print("aws sso login --profile softmax")
-            print("Alternatively, set AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY in your environment.")
-            exit(1)
-    if cfg.wandb.enabled and require_wandb:
-        # Check that W&B is good to go.
-        if not check_wandb_credentials():
-            print("W&B is not configured, please run:")
-            print("wandb login")
-            print("Alternatively, set WANDB_API_KEY or copy ~/.netrc from another machine that has it configured.")
-            exit(1)

mettascope/replays.py

@@ -10,7 +10,6 @@
 from metta.agent.policy_store import PolicyStore
 from metta.sim.simulation import Simulation
 from metta.sim.simulation_config import SingleEnvSimulationConfig
-from metta.util.config import setup_metta_environment
 from metta.util.logging import setup_mettagrid_logger
 from metta.util.runtime_configuration import setup_mettagrid_environment
 from metta.util.wandb.wandb_context import WandbContext
@@ -49,7 +48,6 @@ def policy(self, *args):
 
 
 def create_simulation(cfg):
-    setup_metta_environment(cfg)
     setup_mettagrid_environment(cfg)
 
     logger = setup_mettagrid_logger("replay")

tools/replay.py

@@ -11,7 +11,7 @@
 from metta.agent.policy_store import PolicyStore
 from metta.sim.simulation import Simulation
 from metta.sim.simulation_config import SingleEnvSimulationConfig
-from metta.util.config import Config, setup_metta_environment
+from metta.util.config import Config
 from metta.util.logging import setup_mettagrid_logger
 from metta.util.runtime_configuration import setup_mettagrid_environment
 from metta.util.wandb.wandb_context import WandbContext
@@ -29,7 +29,6 @@ class ReplayJob(Config):
 
 @hydra.main(version_base=None, config_path="../configs", config_name="replay_job")
 def main(cfg):
-    setup_metta_environment(cfg)
     setup_mettagrid_environment(cfg)
 
     logger = setup_mettagrid_logger("metta.tools.replay")

tools/train.py

@@ -13,7 +13,7 @@
 from app_backend.stats_client import StatsClient
 from metta.agent.policy_store import PolicyStore
 from metta.sim.simulation_config import SimulationSuiteConfig
-from metta.util.config import Config, setup_metta_environment
+from metta.util.config import Config
 from metta.util.heartbeat import record_heartbeat
 from metta.util.logging import setup_mettagrid_logger
 from metta.util.runtime_configuration import setup_mettagrid_environment
@@ -69,7 +69,6 @@ def train(cfg: ListConfig | DictConfig, wandb_run: WandbRun | None, logger: Logg
 @record
 @hydra.main(config_path="../configs", config_name="train_job", version_base=None)
 def main(cfg: ListConfig | DictConfig) -> int:
-    setup_metta_environment(cfg)
     setup_mettagrid_environment(cfg)
 
     record_heartbeat()