fix pre-commit

tlambert03 · Jun 30, 2023 · c002ee6 · c002ee6
1 parent fa4548c
commit c002ee6
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 3 deletions.
diff --git a/src/ome_types/_conversion.py b/src/ome_types/_conversion.py
@@ -17,7 +17,7 @@
     from xsdata.formats.dataclass.parsers.mixins import XmlHandler
     from xsdata_pydantic_basemodel.bindings import XmlContext
 
-    from ome_types.model.ome_2016_06 import OME
+    from ome_types.model import OME
 
     class ParserKwargs(TypedDict, total=False):
         config: ParserConfig

diff --git a/src/ome_types/_mixins/_instrument.py b/src/ome_types/_mixins/_instrument.py
@@ -1,7 +1,7 @@
 from typing import TYPE_CHECKING, TypeAlias, Union, cast
 
 if TYPE_CHECKING:
-    from ome_types.model import (
+    from ome_types.model.ome_2016_06 import (
         Arc,
         Filament,
         GenericExcitationSource,

diff --git a/src/ome_types/_mixins/_ome.py b/src/ome_types/_mixins/_ome.py
@@ -8,7 +8,7 @@
 if TYPE_CHECKING:
     from pathlib import Path
 
-    from ome_types.model import OME, Reference
+    from ome_types.model.ome_2016_06 import OME, Reference
 
 
 class OMEMixin:

diff --git a/src/ome_types/model/__init__.py b/src/ome_types/model/__init__.py
@@ -1 +1,5 @@
 from ome_types.model.ome_2016_06 import *  # noqa
+
+# these are here mostly to make mypy happy in pre-commit even when the model isn't built
+from ome_types.model.ome_2016_06 import OME as OME
+from ome_types.model.ome_2016_06 import Reference as Reference