Skip to content

Commit 7bd653d

Browse files
committed
feat: import Sample SQL
1 parent 37ef5c4 commit 7bd653d

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

backend/apps/data_training/curd/data_training.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -416,14 +416,14 @@ def batch_create_training(session: SessionDep, info_list: List[DataTrainingInfo]
416416

417417
try:
418418
# 批量插入
419-
session.bulk_save_objects(data_training_objects)
419+
session.bulk_save_objects(data_training_objects, return_defaults=True)
420420
session.commit()
421421

422-
# 获取插入的ID(需要刷新对象)
422+
# 获取插入的ID
423423
for obj in data_training_objects:
424-
session.refresh(obj)
425-
inserted_ids.append(obj.id)
426-
success_count += 1
424+
if obj.id is not None: # 确保ID已经被赋值
425+
inserted_ids.append(obj.id)
426+
success_count += 1
427427

428428
except Exception as e:
429429
session.rollback()

0 commit comments

Comments
 (0)