put in logic if the model doesn't exist for some reason

SuperCowPowers · Oct 31, 2023 · f3a2622 · f3a2622
1 parent 9b3c153
commit f3a2622
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/scripts/inference_on_endpoint.py b/scripts/inference_on_endpoint.py
@@ -29,7 +29,7 @@
 ENDPOINT_NAME = "stab-regression-end"
 ENDPOINT_NAME = "solubility-test-regression-end"
 ENDPOINT_NAME = "wine-classification-end"
-ENDPOINT_NAME = "abalone-regression-end-rt"
+ENDPOINT_NAME = "abalone-regression-end"
 
 # These should be filled in
 DATA_NAME = "stab_phase_2_features(20)"

diff --git a/src/sageworks/artifacts/endpoints/endpoint.py b/src/sageworks/artifacts/endpoints/endpoint.py
@@ -280,7 +280,10 @@ def model_details(self) -> dict:
             return {}
         else:
             model = Model(self.model_name)
-            return model.details()
+            if model.exists():
+                return model.details()
+            else:
+                return {}
 
     def model_type(self) -> str:
         """Return the type of model used in this Endpoint"""