From d1ad56c748a3f13b1fc652adb112fdabb748099a Mon Sep 17 00:00:00 2001 From: Pascal Pfeiffer Date: Thu, 1 Feb 2024 22:18:14 +0100 Subject: [PATCH] fix: allow null validation rows to be written to db (#605) * fix: allow null validation rows to be written to db * bugfix version version = "1.3.1" --- llm_studio/app_utils/config.py | 2 +- llm_studio/app_utils/db.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/llm_studio/app_utils/config.py b/llm_studio/app_utils/config.py index 2e0b4ebe3..77c30d7a4 100644 --- a/llm_studio/app_utils/config.py +++ b/llm_studio/app_utils/config.py @@ -20,7 +20,7 @@ def get_size(x): return 2**31 -version = "1.4.0-dev" +version = "1.3.1" try: s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) diff --git a/llm_studio/app_utils/db.py b/llm_studio/app_utils/db.py index 1967cbf22..bb41e9622 100644 --- a/llm_studio/app_utils/db.py +++ b/llm_studio/app_utils/db.py @@ -1,3 +1,5 @@ +from typing import Optional + import pandas as pd from pandas.core.frame import DataFrame from sqlalchemy import Integer, String, create_engine @@ -18,7 +20,9 @@ class Dataset(Base): path: Mapped[str] = mapped_column("path", String) config_file: Mapped[str] = mapped_column("config_file", String) train_rows: Mapped[int] = mapped_column("train_rows", Integer) - validation_rows: Mapped[int] = mapped_column("validation_rows", Integer) + validation_rows: Mapped[Optional[int]] = mapped_column( + "validation_rows", Integer, nullable=True + ) class Experiment(Base):