Skip to content

Commit 316f472

Browse files
committed
add validations for update dataset
1 parent 74ee3d8 commit 316f472

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

api/schema/dataset_schema.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -636,7 +636,10 @@ def update_dataset(
636636
dataset = Dataset.objects.get(id=dataset_id)
637637
except Dataset.DoesNotExist as e:
638638
raise ValueError(f"Dataset with ID {dataset_id} does not exist.")
639-
639+
if dataset.status != DatasetStatus.DRAFT.value:
640+
raise ValueError(f"Dataset with ID {dataset_id} is not in draft status.")
641+
if update_dataset_input.title == "":
642+
raise ValueError("Title cannot be empty.")
640643
if update_dataset_input.title:
641644
dataset.title = update_dataset_input.title
642645
if update_dataset_input.description:

0 commit comments

Comments
 (0)