Apply ruff.

Ostorlab · Mar 28, 2024 · f6a8182 · f6a8182
1 parent a28d6a2
commit f6a8182
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git 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
@@ -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"