You've already forked DataMate
fix(annotation): 修改数据集文件状态查询逻辑
- 将文件状态查询从仅统计 ACTIVE 状态扩展为 ACTIVE 和 COMPLETED 状态 - 使用 in_ 操作符替代等于操作符以支持多状态查询 - 保持原有数据集标注计数功能不变
This commit is contained in:
@@ -49,11 +49,11 @@ class DatasetMappingService:
|
||||
Returns:
|
||||
(total_count, annotated_count) 元组
|
||||
"""
|
||||
# 获取数据集总数据量(只统计 ACTIVE 状态的文件)
|
||||
# 获取数据集总数据量(统计 ACTIVE 和 COMPLETED 状态的文件)
|
||||
total_result = await self.db.execute(
|
||||
select(func.count()).select_from(DatasetFiles).where(
|
||||
DatasetFiles.dataset_id == dataset_id,
|
||||
DatasetFiles.status == "ACTIVE",
|
||||
DatasetFiles.status.in_(["ACTIVE", "COMPLETED"]),
|
||||
)
|
||||
)
|
||||
total_count = int(total_result.scalar() or 0)
|
||||
|
||||
Reference in New Issue
Block a user