Silence flake8 warning

RL4AA · Nov 18, 2023 · ad21983 · ad21983
1 parent dcd6319
commit ad21983
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/compare/simon_maml/environment.py b/compare/simon_maml/environment.py
@@ -42,7 +42,7 @@ def generate_optics():
 
 class e_trajectory_simENV(gym.Env):
     def __init__(
-        self, twiss=[], task={}, train=False, **kwargs
+        self, twiss=[], task={}, train=False, **kwargs  # noqa: B006
     ):  # each instance of an environment with new generate_optics()
         self.current_action = None
         self.train = train
@@ -122,7 +122,7 @@ def sample_tasks(self, num_tasks):
         goals = []
         print("Number of tasks: " + str(num_tasks))
         if self.train:
-            for i in range(num_tasks):
+            for i in range(num_tasks):  # noqa: B007
                 twiss = generate_optics()
 
                 twiss_bpms = twiss[twiss["keyword"] == "monitor"]