fixup! Convert branch state management to jobs

Author: mxsrc

src/api/organization/project/branch/control_tasks.py

@@ -80,9 +80,14 @@ async def _async_perform_control(branch_id: str, action: str) -> dict:
     desired_phases = _DESIRED_PHASES[action]
     elapsed = 0.0
     while True:
-        vm = await get_neon_vm(namespace, name)
-        if vm.status is not None and vm.status.phase in desired_phases:
-            break
+        try:
+            vm = await get_neon_vm(namespace, name)
+            if vm.status is not None and vm.status.phase in desired_phases:
+                break
+        except RuntimeError:
+            # VM resource may be transiently absent (404) while NeonVM
+            # reconciles the power-state change; keep polling.
+            logger.debug("VM %r transiently unavailable during %r, retrying", name, action)
         if elapsed >= _TIMEOUT_SEC:
             raise TimeoutError(f"VM {name!r} did not reach {desired_phases} within {_TIMEOUT_SEC}s")
         await asyncio.sleep(_POLL_INTERVAL_SEC)