From a8a9654e635f9173d708075aa7eca99a6fe678c5 Mon Sep 17 00:00:00 2001
From: TShapinsky <tobiasshapinsky@gmail.com>
Date: Fri, 28 Jun 2024 15:11:39 -0600
Subject: [PATCH] fix exception handling for 404 errors

---
 tests/api/conftest.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/api/conftest.py b/tests/api/conftest.py
index e6e995e9..80a11a02 100644
--- 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