velocitatem/PHANTOM
1
0
Fork 0
mirror of https://github.com/velocitatem/PHANTOM.git synced 2026-05-31 08:33:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: cleaning the code

Browse Source
This commit is contained in:
Daniel Rosel
2026-02-28 23:38:38 +01:00
parent 803e3a2972
commit ec880db444
7 changed files with 145 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
engine/jax/train.py
View File

@@ -36,7 +36,7 @@ try:
except ImportError:
HAS_WANDB = False
from ..wandb_checkpoint import (
from ..wandb_checkpoint import ( # noqa: E402
checkpoint_artifact_name,
download_latest_checkpoint,
log_checkpoint_bytes,
@@ -83,7 +83,7 @@ except ImportError:
HAS_JAX_STACK = False
from .env import PHANTOMJAXEnv, make_env_params
from .env import PHANTOMJAXEnv, make_env_params # noqa: E402
class ActorCritic(nn.Module):