1616import asyncio
1717import logging
1818import re
19- from dataclasses import dataclass , field
2019from abc import ABC , abstractmethod
20+ from dataclasses import dataclass , field
2121from enum import Enum , auto
2222from typing import TYPE_CHECKING , Dict , List , Optional , Pattern
2323
@@ -213,7 +213,7 @@ async def execute(
213213 ) # have some requests processed after the rolling upgrade has completed
214214
215215 return await deployment .get_pod_names (self .service_names )
216-
216+
217217 def get_failure_key (self ) -> str :
218218 """Get the failure key for the rolling upgrade failure."""
219219 return f"rolling_upgrade:{ ',' .join (self .service_names )} "
@@ -237,6 +237,8 @@ async def execute(
237237 pod .delete (force = True ) # force means no graceful termination
238238 pod_names .append (pod .name )
239239
240+ return pod_names
241+
240242 def get_failure_key (self ) -> str :
241243 """Get the failure key for the delete pod failure."""
242244 return f"delete_pod:{ ',' .join (self .service_names )} "
@@ -288,8 +290,9 @@ async def execute(
288290 )
289291 process .kill (self .signal )
290292 pod_names .append (pod .name )
293+
291294 return pod_names
292-
295+
293296 def get_failure_key (self ) -> str :
294297 """Get the failure key for the terminate process failure."""
295298 return f"terminate_process:{ ',' .join (self .service_names )} :{ self .process_name } :{ self .signal } "
0 commit comments