Add exception handling

jahaniam · Jan 31, 2024 · 31ce760 · 31ce760
1 parent e36f150
commit 31ce760
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/streamlit_app/app.py b/streamlit_app/app.py
@@ -44,12 +44,15 @@ def get_session():
 
 @st.cache_data
 def get_endpoint_names():
-    sm_client = boto3.client("sagemaker")
-    response = sm_client.list_endpoints(
-        SortBy="CreationTime",
-        SortOrder="Descending",
-    )
-    endpoints = response["Endpoints"]
+    try:
+        sm_client = boto3.client("sagemaker")
+        response = sm_client.list_endpoints(
+            SortBy="CreationTime",
+            SortOrder="Descending",
+        )
+        endpoints = response["Endpoints"]
+    except:
+        endpoints = []
     return endpoints