fix exception handling for 404 errors

NREL · Jun 28, 2024 · 14686b4 · 14686b4
1 parent 8e7fb5a
commit 14686b4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/api/conftest.py b/tests/api/conftest.py
@@ -3,7 +3,7 @@
 
 import pytest
 from alfalfa_client import AlfalfaClient
-from requests import HTTPError
+from alfalfa_client.alfalfa_client import AlfalfaAPIException
 
 
 @pytest.fixture
@@ -33,5 +33,5 @@ def run_id(alfalfa_client: AlfalfaClient, model_path):
     try:
         if alfalfa_client.status(run_id) not in ["COMPLETE", "ERROR", "STOPPING", "READY"]:
             alfalfa_client.stop(run_id)
-    except HTTPError:
+    except AlfalfaAPIException:
         pass