add load_description_and_test overloads for expected_type=dataset

Author: FynnBe

bioimageio/core/_resource_tests.py

@@ -30,8 +30,10 @@
 
 from bioimageio.core import __version__
 from bioimageio.spec import (
+    AnyDatasetDescr,
     AnyModelDescr,
     BioimageioCondaEnv,
+    DatasetDescr,
     InvalidDescr,
     LatestResourceDescr,
     ModelDescr,
@@ -484,6 +486,21 @@ def load_description_and_test(
 ) -> Union[ModelDescr, InvalidDescr]: ...
 
 
+@overload
+def load_description_and_test(
+    source: Union[ResourceDescr, PermissiveFileSource, BioimageioYamlContent],
+    *,
+    format_version: Literal["latest"],
+    weight_format: Optional[SupportedWeightsFormat] = None,
+    devices: Optional[Sequence[str]] = None,
+    determinism: Literal["seed_only", "full"] = "seed_only",
+    expected_type: Literal["dataset"],
+    sha256: Optional[Sha256] = None,
+    stop_early: bool = True,
+    **deprecated: Unpack[DeprecatedKwargs],
+) -> Union[DatasetDescr, InvalidDescr]: ...
+
+
 @overload
 def load_description_and_test(
     source: Union[ResourceDescr, PermissiveFileSource, BioimageioYamlContent],
@@ -514,6 +531,21 @@ def load_description_and_test(
 ) -> Union[AnyModelDescr, InvalidDescr]: ...
 
 
+@overload
+def load_description_and_test(
+    source: Union[ResourceDescr, PermissiveFileSource, BioimageioYamlContent],
+    *,
+    format_version: Union[FormatVersionPlaceholder, str] = DISCOVER,
+    weight_format: Optional[SupportedWeightsFormat] = None,
+    devices: Optional[Sequence[str]] = None,
+    determinism: Literal["seed_only", "full"] = "seed_only",
+    expected_type: Literal["dataset"],
+    sha256: Optional[Sha256] = None,
+    stop_early: bool = True,
+    **deprecated: Unpack[DeprecatedKwargs],
+) -> Union[AnyDatasetDescr, InvalidDescr]: ...
+
+
 @overload
 def load_description_and_test(
     source: Union[ResourceDescr, PermissiveFileSource, BioimageioYamlContent],