@@ -31,19 +31,6 @@ def test_registering_model_pre_upgrade(
3131 errors = get_and_validate_registered_model (
3232 model_registry_client = model_registry_client , model_name = MODEL_NAME , registered_model = registered_model
3333 )
34- # model = model_registry_client.get_registered_model(name=MODEL_NAME)
35- # expected_attrs = {
36- # "id": registered_model.id,
37- # "name": registered_model.name,
38- # "description": registered_model.description,
39- # "owner": registered_model.owner,
40- # "state": registered_model.state,
41- # }
42- # errors = [
43- # f"Unexpected {attr} expected: {expected}, received {getattr(model, attr)}"
44- # for attr, expected in expected_attrs.items()
45- # if getattr(model, attr) != expected
46- # ]
4734 if errors :
4835 pytest .fail ("errors found in model registry response validation:\n {}" .format ("\n " .join (errors )))
4936
@@ -63,15 +50,6 @@ def test_retrieving_model_post_upgrade(
6350 model_registry_client = model_registry_client ,
6451 model_name = MODEL_NAME ,
6552 )
66- # model = model_registry_client.get_registered_model(name=MODEL_NAME)
67- # expected_attrs = {
68- # "name": MODEL_DICT["model_name"],
69- # }
70- # errors = [
71- # f"Unexpected {attr} expected: {expected}, received {getattr(model, attr)}"
72- # for attr, expected in expected_attrs.items()
73- # if getattr(model, attr) != expected
74- # ]
7553 if errors :
7654 pytest .fail ("errors found in model registry response validation:\n {}" .format ("\n " .join (errors )))
7755
0 commit comments