Use relative imports in adapter code

MozillaSecurity · May 29, 2024 · 517c74e · 517c74e
1 parent 63096dd
commit 517c74e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/grizzly/adapter/adapter.py b/grizzly/adapter/adapter.py
@@ -5,11 +5,12 @@
 from pathlib import Path
 from typing import Any, Dict, Generator, Optional, Tuple, final
 
-from grizzly.common.storage import TestCase
-from grizzly.common.utils import DEFAULT_TIME_LIMIT, HARNESS_FILE
-from grizzly.target.target_monitor import TargetMonitor
 from sapphire import ServerMap
 
+from ..common.storage import TestCase
+from ..common.utils import DEFAULT_TIME_LIMIT, HARNESS_FILE
+from ..target.target_monitor import TargetMonitor
+
 __all__ = ("Adapter", "AdapterError")
 __author__ = "Tyson Smith"
 __credits__ = ["Tyson Smith"]

diff --git a/grizzly/adapter/no_op_adapter/__init__.py b/grizzly/adapter/no_op_adapter/__init__.py
@@ -4,10 +4,11 @@
 
 from typing import Optional
 
-from grizzly.adapter import Adapter
-from grizzly.common.storage import TestCase
 from sapphire import ServerMap
 
+from ...common.storage import TestCase
+from ..adapter import Adapter
+
 __author__ = "Tyson Smith"
 __credits__ = ["Tyson Smith"]
 

diff --git a/grizzly/adapter/no_op_adapter/test_no_op.py b/grizzly/adapter/no_op_adapter/test_no_op.py
@@ -2,8 +2,7 @@
 # License, v. 2.0. If a copy of the MPL was not distributed with this
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
-from grizzly.common.storage import TestCase
-
+from ...common.storage import TestCase
 from . import NoOpAdapter