tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
Archived
1
Fork 0
Code Issues Releases Activity

fix: linting

Browse source
This commit is contained in:
Tibo De Peuter 2026-04-09 00:33:56 +02:00
parent 2f31df412b
commit 849581fcfd
Signed by: tdpeuter
SSH key fingerprint: SHA256:u/h/LVoqKF1Iz02uOyxe6hcjmoZASCGV2HM0TG9ZMoU
3 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
scripts/train.py
View file

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

1
src/brittle_star_project/trainers/PPOTrainer.py
View file

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

3
src/experiment_logger/wandb_utils.py
View file

@ -52,7 +52,8 @@ def init_wandb(
if not has_key and not is_interactive and os.environ.get("WANDB_MODE") != "offline": if not has_key and not is_interactive and os.environ.get("WANDB_MODE") != "offline":
logger.warning( 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" sync_path = f"runs/{name}" if name else "runs"
logger.warning(f"WandB is offline. Use 'wandb sync {sync_path}' to upload logs later.") logger.warning(f"WandB is offline. Use 'wandb sync {sync_path}' to upload logs later.")