temporal: Add dataset_factory implementation with str type for type a…

…rgument
echoix · Oct 27, 2024 · 97ddd7c · 97ddd7c
1 parent a263b92
commit 97ddd7c
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/python/grass/temporal/factory.py b/python/grass/temporal/factory.py
@@ -68,6 +68,7 @@ def dataset_factory(type: Literal["vect", "vector"], id: str) -> VectorDataset:
     pass
 
 
+@overload
 def dataset_factory(
     type: Literal[
         "strds",
@@ -82,6 +83,19 @@ def dataset_factory(
         "vector",
     ],
     id: str,
+) -> (
+    SpaceTimeRasterDataset
+    | SpaceTimeRaster3DDataset
+    | SpaceTimeVectorDataset
+    | RasterDataset
+    | Raster3DDataset
+    | VectorDataset
+):
+    pass
+
+
+def dataset_factory(
+    type: str, id: str
 ) -> (
     SpaceTimeRasterDataset
     | SpaceTimeRaster3DDataset