Skip to content

Commit

Permalink
create tables to ensure they exist in the database
Browse files Browse the repository at this point in the history
  • Loading branch information
TilmanNiem committed Jan 28, 2025
1 parent 3d834e0 commit 5849b4e
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/dataland_qa_lab/review/dataset_reviewer.py
Original file line number Diff line number Diff line change
@@ -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
Expand All @@ -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)
Expand Down

0 comments on commit 5849b4e

Please sign in to comment.