1212from bare_utils import set_gh_actions_outputs
1313from bioimageio .spec import load_raw_resource_description , validate
1414from 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
1616from bioimageio .spec .shared import yaml
1717from bioimageio .spec .shared .raw_nodes import Dependencies , URI
1818from 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