diff options
Diffstat (limited to 'test_utils')
| -rw-r--r-- | test_utils/models/network/conftest.py | 4 | ||||
| -rw-r--r-- | test_utils/models/thl/conftest.py | 2 | ||||
| -rw-r--r-- | test_utils/models/upk/conftest.py | 6 |
3 files changed, 4 insertions, 8 deletions
diff --git a/test_utils/models/network/conftest.py b/test_utils/models/network/conftest.py index 4ff59ee..c7bcc7e 100644 --- a/test_utils/models/network/conftest.py +++ b/test_utils/models/network/conftest.py @@ -24,9 +24,7 @@ from generalresearch.models.network.tool_run_command import ( RDNSRunCommand, RDNSRunCommandOptions, ) - -if TYPE_CHECKING: - from generalresearch.pg_helper import PostgresConfig +from generalresearch.pg_helper import PostgresConfig @pytest.fixture(scope="session") diff --git a/test_utils/models/thl/conftest.py b/test_utils/models/thl/conftest.py index 021b19e..f1cb785 100644 --- a/test_utils/models/thl/conftest.py +++ b/test_utils/models/thl/conftest.py @@ -32,6 +32,7 @@ from generalresearch.models.thl.user import User from generalresearch.models.thl.user_iphistory import IPRecord from generalresearch.models.thl.userhealth import AuditLogLevel from generalresearch.models.thl.wallet.definitions import PayoutType +from generalresearch.pg_helper import PostgresConfig if TYPE_CHECKING: from generalresearch.currency import USDCent @@ -73,7 +74,6 @@ if TYPE_CHECKING: from generalresearch.models.thl.user_iphistory import IPRecord from generalresearch.models.thl.userhealth import AuditLog from generalresearch.models.thl.wallet.cashout_method import CashMailOrderData - from generalresearch.pg_helper import PostgresConfig fake = faker.Faker() diff --git a/test_utils/models/upk/conftest.py b/test_utils/models/upk/conftest.py index 59266b2..ad96bbb 100644 --- a/test_utils/models/upk/conftest.py +++ b/test_utils/models/upk/conftest.py @@ -3,15 +3,13 @@ from __future__ import annotations import os import time from collections.abc import Callable -from typing import TYPE_CHECKING from uuid import UUID import pandas as pd import pytest -if TYPE_CHECKING: - from generalresearch.managers.thl.category import CategoryManager - from generalresearch.pg_helper import PostgresConfig +from generalresearch.managers.thl.category import CategoryManager +from generalresearch.pg_helper import PostgresConfig def insert_data_from_csv( |
