aboutsummaryrefslogtreecommitdiff
path: root/test_utils/models
diff options
context:
space:
mode:
Diffstat (limited to 'test_utils/models')
-rw-r--r--test_utils/models/network/conftest.py4
-rw-r--r--test_utils/models/thl/conftest.py2
-rw-r--r--test_utils/models/upk/conftest.py6
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(