Skip to content

Commit d77fc0f

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 775a626 commit d77fc0f

File tree

3 files changed

+10
-3
lines changed

3 files changed

+10
-3
lines changed

mesa_llm/recording/simulation_recorder.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -272,7 +272,9 @@ def record_model_event(self, event_type: str, content: dict[str, Any]):
272272

273273
def get_agent_events(self, agent_id: int) -> list[SimulationEvent]:
274274
"""Get all events for a specific agent."""
275-
return [event for event in self._iter_all_events() if event.agent_id == agent_id]
275+
return [
276+
event for event in self._iter_all_events() if event.agent_id == agent_id
277+
]
276278

277279
def get_events_by_type(self, event_type: str) -> list[SimulationEvent]:
278280
"""Get all events of a specific type."""

tests/test_reasoning/test_decision.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -224,7 +224,10 @@ def test_aplan_async_version(self, llm_response_factory, mock_agent):
224224
{
225225
"name": "move_east",
226226
"description": "Move one cell east",
227-
"tradeoffs": ["Improves coordination", "May increase exposure"],
227+
"tradeoffs": [
228+
"Improves coordination",
229+
"May increase exposure",
230+
],
228231
"score": 0.74,
229232
}
230233
],

tests/test_recording/test_agent_analysis.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,9 @@ def test_init_with_json_file(self, temp_recording_file, sample_recording_data):
2525
assert 123 in viewer.agent_events
2626
assert 456 in viewer.agent_events
2727

28-
def test_init_with_pickle_file_warns(self, temp_recording_file, sample_recording_data):
28+
def test_init_with_pickle_file_warns(
29+
self, temp_recording_file, sample_recording_data
30+
):
2931
"""Test pickle recordings load with a warning."""
3032
_, pkl_path = temp_recording_file
3133

0 commit comments

Comments
 (0)