fix: add missing import for select and DatasetFiles in replace endpoint

jason · jason · commit bff29f4fa358 · 2026-05-06T12:50:05.000+08:00
diff --git a/runtime/datamate-python/app/module/annotation/interface/task.py b/runtime/datamate-python/app/module/annotation/interface/task.py
@@ -351,6 +351,9 @@ async def replace_file_tags(
     """
     service = DatasetManagementService(db)
 
+    from sqlalchemy.future import select
+    from app.db.models import DatasetFiles
+
     result = await db.execute(
         select(DatasetFiles).where(DatasetFiles.id == file_id)
     )

Original file line number	Diff line number	Diff line change
`@@ -351,6 +351,9 @@ async def replace_file_tags(`
`351`	`351`	`"""`
`352`	`352`	`service = DatasetManagementService(db)`
`353`	`353`
	`354`	`+ from sqlalchemy.future import select`
	`355`	`+ from app.db.models import DatasetFiles`
	`356`	`+`
`354`	`357`	`result = await db.execute(`
`355`	`358`	`select(DatasetFiles).where(DatasetFiles.id == file_id)`
`356`	`359`	`)`