Fix last ruff errors

-Use with contextlib.suppress(KeyError) - Supress pickling warning in tests
khughitt · Dec 23, 2023 · 7efc6b6 · 7efc6b6
1 parent 5fead6b
commit 7efc6b6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/mesa/agent.py b/mesa/agent.py
@@ -50,11 +50,8 @@ def __init__(self, unique_id: int, model: Model) -> None:
 
     def remove(self) -> None:
         """Remove and delete the agent from the model."""
-        try:
-            # remove agent
+        with contextlib.suppress(KeyError):
             self.model._agents[type(self)].pop(self)
-        except KeyError:
-            pass
 
     def step(self) -> None:
         """A single step of the agent."""

diff --git a/tests/test_agent.py b/tests/test_agent.py
@@ -82,7 +82,7 @@ def test_function(agent):
     assert agents[0] in agentset
 
     # because AgentSet uses weakrefs, we need hard refs as well....
-    other_agents, another_set = pickle.loads(
+    other_agents, another_set = pickle.loads(  # noqa: S301
         pickle.dumps([agents, AgentSet(agents, model)])
     )
     assert all(