From 1545a85b0992ad58ac174398865d02acb25f6987 Mon Sep 17 00:00:00 2001 From: heimoshuiyu Date: Fri, 18 Oct 2024 19:35:26 +0800 Subject: [PATCH] =?UTF-8?q?=E8=BE=93=E5=87=BA=E9=A3=8E=E9=99=A9=E5=88=86?= =?UTF-8?q?=E6=9E=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cucyuqing/cmd/risk-analyze.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/cucyuqing/cmd/risk-analyze.py b/cucyuqing/cmd/risk-analyze.py index d431e01..749b1f4 100644 --- a/cucyuqing/cmd/risk-analyze.py +++ b/cucyuqing/cmd/risk-analyze.py @@ -118,6 +118,8 @@ async def batch_risk_analyze( task.response = await risk_analyze(task, model) queue.task_done() bar.update(1) + if bar.n % 100 == 0: + print(f"已完成 {bar.n} 条风险分析") async def producer(): for task in tasks: @@ -131,6 +133,8 @@ async def batch_risk_analyze( await queue.put(None) await asyncio.gather(*workers) + print("风险分析完成") + return tasks