def test_search_similar_models(model_args, other_args): model = Zoo.load_model(**model_args, **other_args) similar = Zoo.search_optimized_models(model) assert len(similar) > 0 for sim in similar: assert sim assert sim.domain == model.domain assert sim.sub_domain == model.sub_domain assert sim.architecture == model.architecture assert sim.sub_architecture == model.sub_architecture
def test_search_optimized_models(model_args, other_args): model = Zoo.load_model(**model_args, **other_args) optimized = Zoo.search_optimized_models(model) assert len(optimized) > 0 for sim in optimized: assert sim assert sim.domain == model.domain assert sim.sub_domain == model.sub_domain assert sim.architecture == model.architecture assert sim.sub_architecture == model.sub_architecture assert sim.framework == model.framework assert sim.repo == model.repo assert sim.dataset == model.dataset assert sim.training_scheme == model.training_scheme