aboutsummaryrefslogtreecommitdiff
path: root/tests/models/custom_types
diff options
context:
space:
mode:
Diffstat (limited to 'tests/models/custom_types')
-rw-r--r--tests/models/custom_types/test_aware_datetime.py4
-rw-r--r--tests/models/custom_types/test_uuid_str.py4
2 files changed, 2 insertions, 6 deletions
diff --git a/tests/models/custom_types/test_aware_datetime.py b/tests/models/custom_types/test_aware_datetime.py
index 54a5d9b..e8a5aa3 100644
--- a/tests/models/custom_types/test_aware_datetime.py
+++ b/tests/models/custom_types/test_aware_datetime.py
@@ -2,14 +2,12 @@ from __future__ import annotations
import logging
from datetime import UTC, datetime
-from typing import TYPE_CHECKING
import pytest
import pytz
from pydantic import BaseModel, Field, ValidationError
-if TYPE_CHECKING:
- from generalresearch.models.custom_types import AwareDatetimeISO
+from generalresearch.models.custom_types import AwareDatetimeISO
logger = logging.getLogger()
diff --git a/tests/models/custom_types/test_uuid_str.py b/tests/models/custom_types/test_uuid_str.py
index 92489a0..02e6a8b 100644
--- a/tests/models/custom_types/test_uuid_str.py
+++ b/tests/models/custom_types/test_uuid_str.py
@@ -1,13 +1,11 @@
from __future__ import annotations
-from typing import TYPE_CHECKING
from uuid import uuid4
import pytest
from pydantic import BaseModel, Field, ValidationError
-if TYPE_CHECKING:
- from generalresearch.models.custom_types import UUIDStr
+from generalresearch.models.custom_types import UUIDStr
class UUIDStrModel(BaseModel):