Merge branch 'main' into psi/dpofakepairs

h2oai · Feb 5, 2024 · 33a4ccd · 33a4ccd
2 parents 5c44e9d + 015c752
commit 33a4ccd
Showing 1 changed file with 5 additions and 1 deletion.
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):