add load_description_and_test overloads for expected_type=model

Author: FynnBe

bioimageio/core/_resource_tests.py

@@ -30,9 +30,11 @@
 
 from bioimageio.core import __version__
 from bioimageio.spec import (
+    AnyModelDescr,
     BioimageioCondaEnv,
     InvalidDescr,
     LatestResourceDescr,
+    ModelDescr,
     ResourceDescr,
     ValidationContext,
     build_description,
@@ -467,6 +469,21 @@ def _test_in_env(
     return ValidationSummary.load_json(summary_path)
 
 
+@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["model"],
+    sha256: Optional[Sha256] = None,
+    stop_early: bool = True,
+    **deprecated: Unpack[DeprecatedKwargs],
+) -> Union[ModelDescr, InvalidDescr]: ...
+
+
 @overload
 def load_description_and_test(
     source: Union[ResourceDescr, PermissiveFileSource, BioimageioYamlContent],
@@ -482,6 +499,21 @@ def load_description_and_test(
 ) -> Union[LatestResourceDescr, 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["model"],
+    sha256: Optional[Sha256] = None,
+    stop_early: bool = True,
+    **deprecated: Unpack[DeprecatedKwargs],
+) -> Union[AnyModelDescr, InvalidDescr]: ...
+
+
 @overload
 def load_description_and_test(
     source: Union[ResourceDescr, PermissiveFileSource, BioimageioYamlContent],