diff --git a/tests/unit/test_graph.py b/tests/unit/test_graph.py index d4282bcff95..96e058796b8 100644 --- a/tests/unit/test_graph.py +++ b/tests/unit/test_graph.py @@ -20,12 +20,7 @@ from dbt.graph import NodeSelector, parse_difference from dbt.events.logging import setup_event_logger from dbt.mp_context import get_mp_context - -try: - from queue import Empty -except ImportError: - from Queue import Empty - +from queue import Empty from .utils import config_from_parts_or_dicts, generate_name_macros, inject_plugin diff --git a/tests/unit/test_linker.py b/tests/unit/test_linker.py index 9c36ae19674..973a41c062b 100644 --- a/tests/unit/test_linker.py +++ b/tests/unit/test_linker.py @@ -4,14 +4,9 @@ from unittest import mock from dbt import compilation - -try: - from queue import Empty -except ImportError: - from Queue import Empty - from dbt.graph.selector import NodeSelector from dbt.graph.cli import parse_difference +from queue import Empty def _mock_manifest(nodes):