Containers also working

Author: momchil-flex

docs/notebooks

@@ -389,12 +389,7 @@ def status(self) -> str:
         """Return current status of :class:`Job`."""
         if self.load_if_cached:
             return "success"
-        if web._is_modeler_batch(self.task_id):
-            detail = self.get_info()
-            status = detail.totalStatus.value
-            return status
-        else:
-            return self.get_info().status
+        return self.get_info().status
 
     def start(self, priority: Optional[int] = None) -> None:
         """Start running a :class:`Job`.
@@ -1009,7 +1004,6 @@ def monitor(
         """
         Monitor progress of each running task.
 
-        - For Component Modeler jobs, automatically triggers postprocessing once run finishes.
         - Optionally downloads results as soon as a job reaches final success.
         - Rich progress bars in verbose mode; quiet polling otherwise.
 
@@ -1035,16 +1029,8 @@ def monitor(
             self._check_path_dir(path_dir=path_dir)
             download_executor = ThreadPoolExecutor(max_workers=self.num_workers)
 
-        def _should_download(job: Job) -> bool:
-            status = job.status
-            if not web._is_modeler_batch(job.task_id):
-                return status == "success"
-            if status == "success":
-                return True
-            return status == "run_success" and getattr(job, "postprocess_status", None) == "success"
-
         def schedule_download(job: Job) -> None:
-            if download_executor is None or not _should_download(job):
+            if download_executor is None or job.status not in COMPLETED_STATES:
                 return
             task_id = job.task_id
             if task_id in downloads_started:
@@ -1068,12 +1054,7 @@ def schedule_download(job: Job) -> None:
         def check_continue_condition(job: Job) -> bool:
             if job.load_if_cached:
                 return False
-            status = job.status
-            if not web._is_modeler_batch(job.task_id):
-                return status not in END_STATES
-            if status == "run_success":
-                return job.postprocess_status not in END_STATES
-            return status not in END_STATES
+            return job.status not in END_STATES
 
         def pbar_description(
             task_name: str, status: str, max_name_length: int, status_width: int
@@ -1097,9 +1078,6 @@ def pbar_description(
         max_task_name = max(len(task_name) for task_name in self.jobs.keys())
         max_name_length = min(30, max(max_task_name, 15))
 
-        # track which modeler jobs we've already kicked into postprocess
-        postprocess_started_tasks: set[str] = set()
-
         try:
             console = None
             progress_columns = []
@@ -1135,17 +1113,6 @@ def pbar_description(
                     for task_name, job in self.jobs.items():
                         status = job.status
 
-                        # auto-start postprocess for modeler jobs when run finishes
-                        if (
-                            web._is_modeler_batch(job.task_id)
-                            and status == "run_success"
-                            and job.task_id not in postprocess_started_tasks
-                        ):
-                            job.postprocess_start(
-                                worker_group=postprocess_worker_group, verbose=True
-                            )
-                            postprocess_started_tasks.add(job.task_id)
-
                         schedule_download(job)
 
                         if self.verbose:

tidy3d/web/api/container.py

@@ -5,15 +5,16 @@
 }
 
 ERROR_STATES = {
-    "validate_fail",
+    "validate_error",
     "error",
     "errored",
     "diverge",
     "diverged",
     "blocked",
-    "run_failed",
+    "run_error",
     "aborted",
     "deleted",
+    "postprocess_error",
 }
 
 PRE_VALIDATE_STATES = {
@@ -30,7 +31,7 @@
     "run_success",
 }
 
-COMPLETED_STATES = {"visualize", "success", "completed", "processed"}
+COMPLETED_STATES = {"visualize", "success", "completed", "processed", "postprocess_success"}
 
 END_STATES = ERROR_STATES | COMPLETED_STATES
 

tidy3d/web/api/states.py

@@ -123,7 +123,7 @@ def _batch_detail_error(resource_id: str) -> Optional[WebError]:
         return None
 
     if (
-        status == "validate_fail"
+        status in ERROR_STATES
         and hasattr(batch_detail, "validateErrors")
         and batch_detail.validateErrors
     ):
@@ -1511,8 +1511,9 @@ def estimate_cost(
 
     if _is_modeler_batch(task_id):
         batch = BatchTask(task_id)
-        resp = batch.check(solver_version=solver_version)
-        status = resp["validateStatus"]
+        _ = batch.check(solver_version=solver_version)
+        detail = batch.detail()
+        status = detail.status.lower()
         while status not in ALL_POST_VALIDATE_STATES:
             time.sleep(REFRESH_TIME)
             detail = batch.detail()