Merge pull request #6 from WaveSpeedAI/dev

Add feature

Author: joeliu926

src/wavespeed/api/client.py

@@ -183,11 +183,6 @@ def _submit(
                         f"Failed to submit prediction after {self.max_connection_retries + 1} attempts"
                     ) from e
 
-        # Should not reach here, but guard against it
-        raise last_error or RuntimeError(
-            f"Failed to submit prediction after {self.max_connection_retries + 1} attempts"
-        )
-
     def _get_result(
         self, request_id: str, timeout: float | None = None
     ) -> dict[str, Any]:
@@ -265,12 +260,6 @@ def _get_result(
                         f"after {self.max_connection_retries + 1} attempts"
                     ) from e
 
-        # Should not reach here, but guard against it
-        raise last_error or RuntimeError(
-            f"Failed to get result for task {request_id} "
-            f"after {self.max_connection_retries + 1} attempts"
-        )
-
     def _wait(
         self,
         request_id: str,