add #noqa to fix pre-commit

vwxyzjn · Sep 10, 2024 · 2b13995 · 2b13995
1 parent e9c2f54
commit 2b13995
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
+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
+from pom_env import PoMEnv # noqa
 from torch.distributions import Categorical
 from torch.utils.tensorboard import SummaryWriter