Type annotation tweaks

rearc · Jun 3, 2024 · d969cb7 · d969cb7
1 parent 74c846b
commit d969cb7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_basics.py b/tests/test_basics.py
@@ -1,5 +1,7 @@
 import abc
 import uuid
+from typing import Optional
+from typing import Type
 
 import pytest
 
@@ -21,7 +23,7 @@ class User(_BaseTableModel):
     __table_name__ = "user"
     __str_id_field__ = "name"
     name: str
-    group_id: uuid.UUID | None = None
+    group_id: Optional[uuid.UUID] = None
 
 
 class Group(_BaseTableModel):
@@ -31,7 +33,7 @@ class Group(_BaseTableModel):
 
 
 @pytest.fixture(scope="function", autouse=True)
-def recreate_pynamodb_table() -> type[SingleTableBaseModel]:
+def recreate_pynamodb_table() -> Type[SingleTableBaseModel]:
     _BaseTableModel.__pynamodb_model__.create_table(
         wait=True, billing_mode="PAY_PER_REQUEST"
     )