Skip to content

Commit 199e8d7

Browse files
committed
pass args to session execute
1 parent 89846fd commit 199e8d7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ydb/query/pool.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,10 +123,10 @@ def execute_with_retries(
123123
with self.checkout() as session:
124124

125125
def wrapped_callee():
126-
it = session.execute(query, empty_tx_control=True)
126+
it = session.execute(query, empty_tx_control=True, *args, **kwargs)
127127
return [result_set for result_set in it]
128128

129-
opt_generator = retry_operation_impl(wrapped_callee, retry_settings, *args, **kwargs)
129+
opt_generator = retry_operation_impl(wrapped_callee, retry_settings)
130130
for next_opt in opt_generator:
131131
if isinstance(next_opt, YdbRetryOperationSleepOpt):
132132
time.sleep(next_opt.timeout)

0 commit comments

Comments
 (0)