Skip to content

Commit b9492a5

Browse files
committed
adapt static_validation script to latest spec release
1 parent 60fa000 commit b9492a5

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

scripts/static_validation.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from bare_utils import set_gh_actions_outputs
1313
from bioimageio.spec import load_raw_resource_description, validate
1414
from bioimageio.spec.model.raw_nodes import Model, WeightsFormat
15-
from bioimageio.spec.rdf.raw_nodes import RDF
15+
from bioimageio.spec.rdf.raw_nodes import RDF_Base
1616
from bioimageio.spec.shared import yaml
1717
from bioimageio.spec.shared.raw_nodes import Dependencies, URI
1818
from utils import ADJECTIVES, ANIMALS, iterate_over_gh_matrix, split_animal_nickname
@@ -134,9 +134,7 @@ def ensure_valid_conda_env_name(name: str) -> str:
134134
return name or "empty"
135135

136136

137-
def prepare_dynamic_test_cases(
138-
rd: Union[Model, RDF], resource_id: str, version_id: str, dist: Path
139-
) -> List[Dict[str, str]]:
137+
def prepare_dynamic_test_cases(rd: RDF_Base, resource_id: str, version_id: str, dist: Path) -> List[Dict[str, str]]:
140138
validation_cases = []
141139
# construct test cases based on resource type
142140
if isinstance(rd, Model):
@@ -158,7 +156,7 @@ def prepare_dynamic_test_cases(
158156
validation_cases.append(
159157
{"env_name": env_name, "resource_id": resource_id, "version_id": version_id, "weight_format": wf}
160158
)
161-
elif isinstance(rd, RDF):
159+
elif isinstance(rd, RDF_Base):
162160
pass
163161
else:
164162
raise TypeError(rd)
@@ -211,7 +209,7 @@ def main(
211209
latest_static_summary = validate(rdf_path, update_format=True)
212210
if not latest_static_summary["error"]:
213211
rd = load_raw_resource_description(rdf_path, update_to_format="latest")
214-
assert isinstance(rd, RDF)
212+
assert isinstance(rd, RDF_Base)
215213
dynamic_test_cases += prepare_dynamic_test_cases(rd, resource_id, version_id, dist)
216214

217215
if "name" not in latest_static_summary:

0 commit comments

Comments
 (0)