Patch exit on completed state

Author: daquinteroflex

tidy3d/web/api/webapi.py

@@ -1132,10 +1132,11 @@ def _status_to_stage(status: str) -> tuple[str, int]:
         while True:
             detail = _batch_detail(batch_id)
             status = detail.totalStatus.value
+            postprocess_status = detail.postprocessStatus
             total = detail.totalTask or 0
             p = detail.postprocessSuccess or 0
             progress.update(p_post, completed=(p / total) if total else 0.0)
-            if total and p >= total:
+            if postprocess_status == "completed":
                 break
             if status in terminal_errors:
                 raise WebError(f"Batch {batch_id} terminated: {status}")

tidy3d/web/core/task_info.py

@@ -232,7 +232,7 @@ class BatchDetail(TaskBase):
     totalStatus: BatchStatus = None
     totalTask: int = 0
     preprocessSuccess: int = 0
-    postprocessStatus: TaskStatus = None
+    postprocessStatus: str = None
     validateSuccess: int = 0
     runSuccess: int = 0
     postprocessSuccess: int = 0