resolve flake8 pre commits

adaptive-intelligent-robotics · Mar 10, 2023 · c8d0c2f · c8d0c2f
1 parent 98ed194
commit c8d0c2f
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 6 deletions.
diff --git a/qdax/core/containers/mapelites_repertoire.py b/qdax/core/containers/mapelites_repertoire.py
@@ -343,7 +343,8 @@ def init(
             (
                 "This type of repertoire does not store the extra scores "
                 "computed by the scoring function"
-            )
+            ),
+            stacklevel=2,
         )
 
         # retrieve one genotype from the population

diff --git a/qdax/core/containers/mome_repertoire.py b/qdax/core/containers/mome_repertoire.py
@@ -389,7 +389,8 @@ def init(  # type: ignore
             (
                 "This type of repertoire does not store the extra scores "
                 "computed by the scoring function"
-            )
+            ),
+            stacklevel=2,
         )
 
         # get dimensions

diff --git a/qdax/environments/exploration_wrappers.py b/qdax/environments/exploration_wrappers.py
@@ -151,7 +151,10 @@ def __init__(self, env: env.Env, env_name: str) -> None:
         ):
             raise NotImplementedError(f"This wrapper does not support {env_name} yet.")
         if env_name not in ["ant", "humanoid"]:
-            warnings.warn("Make sure your agent can move in two dimensions!")
+            warnings.warn(
+                "Make sure your agent can move in two dimensions!",
+                stacklevel=2,
+            )
         super().__init__(env)
         self._env_name = env_name
         # update the env config to add the trap
@@ -367,9 +370,14 @@ def __init__(self, env: env.Env, env_name: str) -> None:
             or env_name not in COG_NAMES.keys()
             or env_name not in ENV_MAZE_COLLISION.keys()
         ):
-            raise NotImplementedError(f"This wrapper does not support {env_name} yet.")
+            raise NotImplementedError(
+                f"This wrapper does not support {env_name} yet.",
+            )
         if env_name not in ["ant", "humanoid"]:
-            warnings.warn("Make sure your agent can move in two dimensions!")
+            warnings.warn(
+                "Make sure your agent can move in two dimensions!",
+                stacklevel=2,
+            )
         super().__init__(env)
         self._env_name = env_name
         self._config = (

diff --git a/tests/utils_test/metrics_test.py b/tests/utils_test/metrics_test.py
@@ -13,7 +13,7 @@ def test_csv_logger() -> None:
         csv_logger.log(metrics)
 
     metrics = {"bad_name": 0.0, "max_fitness": 0.0, "coverage": 0.0}
-    with pytest.raises(Exception):
+    with pytest.raises(Exception):  # noqa: B017
         csv_logger.log(metrics)
 
     file_exist = os.path.exists(file_location)