Reuse obstore.store type hinting (#38)

developmentseed · Feb 28, 2025 · 937ec7b · 937ec7b
1 parent 483327b
commit 937ec7b
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 3 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -0,0 +1,3 @@
+[submodule "python/obstore"]
+	path = python/_obstore
+	url = https://github.com/developmentseed/obstore
diff --git a/python/_obstore b/python/_obstore
diff --git a/python/python/async_tiff/__init__.py b/python/python/async_tiff/__init__.py
@@ -1,4 +1,9 @@
+from typing import TYPE_CHECKING
+
 from ._async_tiff import *
 from ._async_tiff import ___version
 
+if TYPE_CHECKING:
+    from . import store
+
 __version__: str = ___version()
diff --git a/python/python/async_tiff/_tiff.pyi b/python/python/async_tiff/_tiff.pyi
@@ -1,11 +1,10 @@
-from typing import Any
-
 from ._ifd import ImageFileDirectory
+from .store import ObjectStore
 
 class TIFF:
     @classmethod
     async def open(
-        cls, path: str, *, store: Any, prefetch: int | None = 16384
+        cls, path: str, *, store: ObjectStore, prefetch: int | None = 16384
     ) -> TIFF: ...
     @property
     def ifds(self) -> list[ImageFileDirectory]: ...
diff --git a/python/python/async_tiff/store b/python/python/async_tiff/store
@@ -0,0 +1 @@
+../../_obstore/obstore/python/obstore/store