feat(api): minor try catch change

AntaresSimulatorTeam · Jan 10, 2025 · 0e6896f · 0e6896f
1 parent cc13286
commit 0e6896f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/src/antares/craft/service/api_services/study_api.py b/src/antares/craft/service/api_services/study_api.py
@@ -24,7 +24,7 @@
     StudySettingsUpdateError,
     StudyVariantCreationError,
     TaskFailedError,
-    ThermalTimeseriesGenerationError,
+    ThermalTimeseriesGenerationError, TaskTimeOutError,
 )
 from antares.craft.model.binding_constraint import BindingConstraint
 from antares.craft.model.output import Output
@@ -172,5 +172,5 @@ def generate_thermal_timeseries(self) -> None:
             response = self._wrapper.put(url)
             task_id = response.json()
             wait_task_completion(self._base_url, self._wrapper, task_id)
-        except (APIError, TaskFailedError) as e:
+        except (APIError, TaskFailedError, TaskTimeOutError) as e:
             raise ThermalTimeseriesGenerationError(self.study_id, e.message)
diff --git a/tests/integration/test_web_client.py b/tests/integration/test_web_client.py
@@ -626,9 +626,4 @@ def test_creation_lifecycle(self, antares_web: AntaresWebDesktop):
         # delete_outputs
         study.delete_outputs()
         assert len(study.get_outputs()) == 0
-        assert len(study.read_outputs()) == 0
-
-        # areas = study.get_areas()
-        # for area_id, area in areas.items():
-        #     for cluster in area.get_thermals().values():
-        #         area._thermal_service.update_thermal_properties(cluster, ThermalClusterProperties(nominal_capacity=1000))
+        assert len(study.read_outputs()) == 0