From 5849b4efc31a75b34216909271bd1f733b8f312a Mon Sep 17 00:00:00 2001 From: TilmanNiem Date: Tue, 28 Jan 2025 16:40:28 +0100 Subject: [PATCH] create tables to ensure they exist in the database --- src/dataland_qa_lab/review/dataset_reviewer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/dataland_qa_lab/review/dataset_reviewer.py b/src/dataland_qa_lab/review/dataset_reviewer.py index 5621252..20cf8ab 100644 --- a/src/dataland_qa_lab/review/dataset_reviewer.py +++ b/src/dataland_qa_lab/review/dataset_reviewer.py @@ -1,6 +1,6 @@ from datetime import UTC, datetime, timedelta, timezone -from dataland_qa_lab.database.database_engine import add_entity, get_entity, update_entity +from dataland_qa_lab.database.database_engine import add_entity, create_tables, get_entity, update_entity from dataland_qa_lab.database.database_tables import ReviewedDataset from dataland_qa_lab.dataland import dataset_provider from dataland_qa_lab.pages import pages_provider, text_to_doc_intelligence @@ -13,6 +13,8 @@ def review_dataset(data_id: str) -> str | None: """Review a dataset.""" dataset = dataset_provider.get_dataset_by_id(data_id) + create_tables() + existing_entity = get_entity(data_id, ReviewedDataset) now_utc = datetime.now(UTC)