Merge pull request #148 from opendatalab/feat/json-pre-annotation

fix: update alembic script
opendatalab · Nov 19, 2024 · 91e70d7 · 91e70d7
2 parents 456ab2c + 97d6c08
commit 91e70d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/labelu/alembic_labelu/versions/eb9c5b98168b_add_sample_name_and_data_to_task_pre_.py b/labelu/alembic_labelu/versions/eb9c5b98168b_add_sample_name_and_data_to_task_pre_.py
@@ -97,7 +97,7 @@ def upgrade() -> None:
                     new_task_pre_annotation = TaskPreAnnotation(
                         task_id=task_pre_annotation.task_id,
                         # full file name
-                        sample_name=sample_file.filename,
+                        sample_name=sample_file.filename if sample_file else None,
                         file_id=file_id,
                         created_by=task_pre_annotation.created_by,
                         updated_by=task_pre_annotation.updated_by,