From f6a8182c63198c68e33f1c2608768378abe217b5 Mon Sep 17 00:00:00 2001
From: mohsinenar <mnarmouche@gmail.com>
Date: Thu, 28 Mar 2024 14:33:00 +0000
Subject: [PATCH] Apply ruff.

---
 agent/nebula_agent.py      |  6 +++---
 tests/nebula_agent_test.py | 18 +++++++++---------
 2 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/agent/nebula_agent.py b/agent/nebula_agent.py
index 648d856..15f772e 100644
--- a/agent/nebula_agent.py
+++ b/agent/nebula_agent.py
@@ -34,9 +34,9 @@ class NebulaAgent(agent.Agent):
     """Agent responsible for persisting all types of messages to file type specified in the agent definition."""
 
     def __init__(
-            self,
-            agent_definition: agent_definitions.AgentDefinition,
-            agent_settings: runtime_definitions.AgentSettings,
+        self,
+        agent_definition: agent_definitions.AgentDefinition,
+        agent_settings: runtime_definitions.AgentSettings,
     ) -> None:
         super().__init__(agent_definition, agent_settings)
         self._file_type = self.args.get("file_type", "json")
diff --git a/tests/nebula_agent_test.py b/tests/nebula_agent_test.py
index 6f1c5de..9f49748 100644
--- a/tests/nebula_agent_test.py
+++ b/tests/nebula_agent_test.py
@@ -37,9 +37,9 @@ def testAgentNebula_whenUnsupportedFileType_raisesValueError() -> None:
 
 
 def testAgentNebula_whenFileTypeIsJson_persistMessage(
-        agent_definition: agent_definitions.AgentDefinition,
-        agent_settings: runtime_definitions.AgentSettings,
-        link_message: msg.Message,
+    agent_definition: agent_definitions.AgentDefinition,
+    agent_settings: runtime_definitions.AgentSettings,
+    link_message: msg.Message,
 ) -> None:
     """Test that NebulaAgent persists message to json file."""
     os.environ["UNIVERSE"] = "43"
@@ -61,9 +61,9 @@ def testAgentNebula_whenFileTypeIsJson_persistMessage(
 
 
 def testAgentNebula_whenFileTypeIsJson_persistMultipleLinkMessages(
-        agent_definition: agent_definitions.AgentDefinition,
-        agent_settings: runtime_definitions.AgentSettings,
-        multiple_link_messages: list[msg.Message],
+    agent_definition: agent_definitions.AgentDefinition,
+    agent_settings: runtime_definitions.AgentSettings,
+    multiple_link_messages: list[msg.Message],
 ) -> None:
     """Test that NebulaAgent persists multiple link messages to json file."""
     os.environ["UNIVERSE"] = "43"
@@ -91,9 +91,9 @@ def testAgentNebula_whenFileTypeIsJson_persistMultipleLinkMessages(
 
 
 def testAgentNebula_whenFileTypeIsJson_persistMultipleMessages(
-        agent_definition: agent_definitions.AgentDefinition,
-        agent_settings: runtime_definitions.AgentSettings,
-        multiple_messages: list[msg.Message],
+    agent_definition: agent_definitions.AgentDefinition,
+    agent_settings: runtime_definitions.AgentSettings,
+    multiple_messages: list[msg.Message],
 ) -> None:
     """Test that NebulaAgent persists multiple messages of different types to json files."""
     os.environ["UNIVERSE"] = "43"