Apply ruff.

agent/nebula_agent.py

@@ -4,7 +4,6 @@
 import json
 import logging
 import os
-from datetime import datetime
 from typing import Any
 
 from ostorlab.agent import agent, definitions as agent_definitions
@@ -35,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")

tests/nebula_agent_test.py

@@ -5,15 +5,13 @@
 import pathlib
 
 import pytest
-from freezegun import freeze_time
 from ostorlab.agent import definitions as agent_definitions
 from ostorlab.agent.message import message as msg
 from ostorlab.runtimes import definitions as runtime_definitions
 from ostorlab.utils import defintions as utils_definitions
 from pyfakefs import fake_filesystem_unittest
 
 from agent import nebula_agent
-import os
 
 
 def testAgentNebula_whenUnsupportedFileType_raisesValueError() -> None:
@@ -39,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"
@@ -63,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"
@@ -93,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"