fix weakref type

tlambert03 · Jul 1, 2023 · c3f669d · c3f669d
1 parent f74de81
commit c3f669d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/ome_types/_mixins/_reference.py b/src/ome_types/_mixins/_reference.py
@@ -1,13 +1,11 @@
-from typing import TYPE_CHECKING, Any, Optional
+import weakref
+from typing import Any, Optional
 
 from ._base_type import OMEType
 
-if TYPE_CHECKING:
-    import weakref
-
 
 class ReferenceMixin(OMEType):
-    _ref: Optional["weakref.ReferenceType[OMEType]"] = None
+    _ref: Optional[weakref.ReferenceType] = None
 
     @property
     def ref(self) -> "OMEType | None":

diff --git a/tests/test_names.py b/tests/test_names.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 from pathlib import Path
 from typing import Any, Sequence