fix(tests): fix deploy_model e2e test (#360)

IBM · May 28, 2024 · 5e3e7f9 · 5e3e7f9
1 parent 1e6d20e
commit 5e3e7f9
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/examples/deployment/deployed_models.py b/examples/deployment/deployed_models.py
@@ -29,8 +29,7 @@ def heading(text: str) -> str:
 
 if len(deployment_list.results) < 1:
     print("No deployed models found.")
-    exit(1)
-
-print(heading("Retrieve information about first deployment"))
-deployment_info = client.deployment.retrieve(id=deployment_list.results[0].id)
-pprint(deployment_info.model_dump())
+else:
+    print(heading("Retrieve information about first deployment"))
+    deployment_info = client.deployment.retrieve(id=deployment_list.results[0].id)
+    pprint(deployment_info.model_dump())