@@ -56,6 +56,12 @@ def _create_and_get_database_details(
5656 )
5757
5858
59+ @retry (
60+ retry = retry_if_exception_type (exception_types = TimeoutException ),
61+ stop = stop_after_attempt (max_attempt_number = 3 ),
62+ wait = wait_exponential (multiplier = 2 , min = 5 , max = 30 ),
63+ reraise = True ,
64+ )
5965def _create_and_get_vumark_details (
6066 driver : "WebDriver" ,
6167 vumark_database_name : str ,
@@ -100,23 +106,6 @@ def _create_and_get_database_details_with_retries(
100106 )
101107
102108
103- @retry (
104- retry = retry_if_exception_type (exception_types = TimeoutException ),
105- stop = stop_after_attempt (max_attempt_number = 3 ),
106- wait = wait_exponential (multiplier = 2 , min = 5 , max = 30 ),
107- reraise = True ,
108- )
109- def _create_and_get_vumark_details_with_retries (
110- driver : "WebDriver" ,
111- vumark_database_name : str ,
112- ) -> "VuMarkDatabaseDict" :
113- """Create a VuMark database and return details with retries on timeout."""
114- return _create_and_get_vumark_details (
115- driver = driver ,
116- vumark_database_name = vumark_database_name ,
117- )
118-
119-
120109def _generate_secrets_file_content (
121110 database_details : "DatabaseDict" ,
122111 vumark_details : "VuMarkDatabaseDict" ,
@@ -195,7 +184,7 @@ def main() -> None:
195184 continue
196185
197186 try :
198- vumark_details = _create_and_get_vumark_details_with_retries (
187+ vumark_details = _create_and_get_vumark_details (
199188 driver = driver ,
200189 vumark_database_name = vumark_database_name ,
201190 )
0 commit comments