From c8d0c2fc14f22b849deb32901c21c3c5a2189f05 Mon Sep 17 00:00:00 2001
From: Felix <f.chalumeau@instadeep.com>
Date: Fri, 10 Mar 2023 11:32:28 +0200
Subject: [PATCH] resolve flake8 pre commits

---
 qdax/core/containers/mapelites_repertoire.py |  3 ++-
 qdax/core/containers/mome_repertoire.py      |  3 ++-
 qdax/environments/exploration_wrappers.py    | 14 +++++++++++---
 tests/utils_test/metrics_test.py             |  2 +-
 4 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/qdax/core/containers/mapelites_repertoire.py b/qdax/core/containers/mapelites_repertoire.py
index 7a24d826..1babb2c9 100644
--- 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
index 68fd5859..d1da327a 100644
--- 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
index b6754c28..33428994 100644
--- 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
index ef4b4488..8aa33494 100644
--- 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)