fix pre-commit #noqa

vwxyzjn · Sep 10, 2024 · 10ecb84 · 10ecb84
1 parent 2b13995
commit 10ecb84
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cleanrl/ppo_trxl/ppo_trxl.py b/cleanrl/ppo_trxl/ppo_trxl.py
@@ -5,15 +5,15 @@
 from dataclasses import dataclass
 
 import gymnasium as gym
-import memory_gym # noqa
+import memory_gym  # noqa
 import numpy as np
 import torch
 import torch.nn as nn
 import torch.optim as optim
 import tyro
 from einops import rearrange
 from minigrid.wrappers import ImgObsWrapper, RGBImgPartialObsWrapper
-from pom_env import PoMEnv # noqa
+from pom_env import PoMEnv  # noqa
 from torch.distributions import Categorical
 from torch.utils.tensorboard import SummaryWriter