Fix 404 handling

challenge-cli.py

@@ -7,8 +7,9 @@ from datetime import datetime, timedelta, timezone
 from pathlib import Path
 
 import click
+import httpx
 
-from dyff.client import Client
+from dyff.client import Client, errors
 from dyff.schema.platform import *
 from dyff.schema.requests import *
 
@@ -28,14 +29,11 @@ def _wait_for_status(
             status = get_entity_fn().status
             if status in target_status:
                 return status
-        except Exception as ex:
-            if hasattr(ex, "status"):
-                if ex.status != 404:  # pyright: ignore
-                    raise
-            elif hasattr(ex, "status_code"):
-                if ex.status_code != 404:  # pyright: ignore
-                    raise
-            else:
+        except errors.HTTPError as ex:
+            if ex.status != 404:
+                raise
+        except httpx.HTTPStatusError as ex:
+            if ex.response.status_code != 404:
                 raise
         if (datetime.now(timezone.utc) - then) >= timeout:
             break