diff --git a/src/runloop_api_client/resources/devboxes/devboxes.py b/src/runloop_api_client/resources/devboxes/devboxes.py index 2a3bcbafe..5657965c0 100644 --- a/src/runloop_api_client/resources/devboxes/devboxes.py +++ b/src/runloop_api_client/resources/devboxes/devboxes.py @@ -398,7 +398,7 @@ def wait_for_devbox_status() -> DevboxView: # If it's neither, it will throw an error. return self._post( f"/v1/devboxes/{id}/wait_for_status", - body={"statuses": ["running", "failure"]}, + body={"statuses": ["running", "failure", "shutdown"]}, cast_to=DevboxView, ) @@ -1752,7 +1752,7 @@ async def wait_for_devbox_status() -> DevboxView: try: return await self._post( f"/v1/devboxes/{id}/wait_for_status", - body={"statuses": ["running", "failure"]}, + body={"statuses": ["running", "failure", "shutdown"]}, cast_to=DevboxView, ) except (APITimeoutError, APIStatusError) as error: