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"
h2oai · Feb 1, 2024 · d1ad56c · d1ad56c
1 parent 2c83849
commit d1ad56c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git 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
@@ -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):