fix: linting

scripts/train.py

@@ -2,8 +2,6 @@ import subprocess
 import time
 
 import torch
-import tyro
-import yaml
 import os
 
 from brittle_star_project.dataclasses import PPOArgs

src/brittle_star_project/trainers/PPOTrainer.py

@@ -5,7 +5,6 @@ from dataclasses import asdict, dataclass
 from functools import partial
 from typing import Any
 
-import flax
 import jax
 import jax.numpy as jnp
 import numpy as np

src/experiment_logger/wandb_utils.py

@@ -52,7 +52,8 @@ def init_wandb(
 
         if not has_key and not is_interactive and os.environ.get("WANDB_MODE") != "offline":
             logger.warning(
-                "WANDB_API_KEY not found and environment is non-interactive. Switching to offline mode."
+                "WANDB_API_KEY not found and environment is non-interactive. "
+                "Switching to offline mode."
             )
             sync_path = f"runs/{name}" if name else "runs"
             logger.warning(f"WandB is offline. Use 'wandb sync {sync_path}' to upload logs later.")