Skip to content

Commit e93d9ab

Browse files
fix: lag for decrypt
1 parent d3caf58 commit e93d9ab

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/rip.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,7 @@ async def decrypt_sample_with_retry(self, adam_id: str, key: str, sample: bytes,
340340
await it(WrapperManager).decrypt(adam_id, key, sample, sample_index)
341341

342342
# Wait for the future to be resolved by the callback
343-
return await future
343+
return await asyncio.wait_for(future, timeout=10)
344344

345345
async def on_decrypt_success(self, adam_id: str, key: str, sample: bytes, sample_index: int):
346346
it(Measurer).record_decrypt(len(sample))

0 commit comments

Comments
 (0)