[HWORKS-1805] Increase timeouts for starting/stopping deployments (lo…

…gicalclocks#399)

python/hsml/deployment.py

@@ -67,7 +67,7 @@ def __init__(
         self._model_registry_id = None
 
     @usage.method_logger
-    def save(self, await_update: Optional[int] = 60):
+    def save(self, await_update: Optional[int] = 120):
         """Persist this deployment including the predictor and metadata to Model Serving.
 
         # Arguments
@@ -79,7 +79,7 @@ def save(self, await_update: Optional[int] = 60):
         self._serving_engine.save(self, await_update)
 
     @usage.method_logger
-    def start(self, await_running: Optional[int] = 60):
+    def start(self, await_running: Optional[int] = 120):
         """Start the deployment
 
         # Arguments
@@ -91,7 +91,7 @@ def start(self, await_running: Optional[int] = 60):
         self._serving_engine.start(self, await_status=await_running)
 
     @usage.method_logger
-    def stop(self, await_stopped: Optional[int] = 60):
+    def stop(self, await_stopped: Optional[int] = 120):
         """Stop the deployment
 
         # Arguments

python/tests/test_deployment.py

@@ -145,7 +145,7 @@ def test_save_default(self, mocker, backend_fixtures):
         d.save()
 
         # Assert
-        mock_serving_engine_save.assert_called_once_with(d, 60)
+        mock_serving_engine_save.assert_called_once_with(d, 120)
 
     def test_save(self, mocker, backend_fixtures):
         # Arrange
@@ -176,7 +176,7 @@ def test_start_default(self, mocker, backend_fixtures):
         d.start()
 
         # Assert
-        mock_serving_engine_start.assert_called_once_with(d, await_status=60)
+        mock_serving_engine_start.assert_called_once_with(d, await_status=120)
 
     def test_start(self, mocker, backend_fixtures):
         # Arrange
@@ -207,7 +207,7 @@ def test_stop_default(self, mocker, backend_fixtures):
         d.stop()
 
         # Assert
-        mock_serving_engine_stop.assert_called_once_with(d, await_status=60)
+        mock_serving_engine_stop.assert_called_once_with(d, await_status=120)
 
     def test_stop(self, mocker, backend_fixtures):
         # Arrange