diff --git a/cleanrl/ppo_trxl/ppo_trxl.py b/cleanrl/ppo_trxl/ppo_trxl.py
index ab37703c..20332be4 100644
--- a/cleanrl/ppo_trxl/ppo_trxl.py
+++ b/cleanrl/ppo_trxl/ppo_trxl.py
@@ -5,7 +5,7 @@
 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
@@ -13,7 +13,7 @@
 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