Add dependency for SQLAlchemy.

ZettaAI · Jan 10, 2025 · 0f5f664 · 0f5f664
1 parent 86dd651
commit 0f5f664
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -162,6 +162,9 @@ skeletonization = [
 slurm = [
   "simple_slurm",
 ]
+sql = [
+   "SQLAlchemy >= 2.0.36",
+]
 tensor-ops = [
   "zetta_utils[tensor_typing]",
   "tinybrain",

diff --git a/scripts/cave_db_sql.py b/scripts/cave_db_sql.py
@@ -66,7 +66,7 @@ def format_value(value: Any, column_name: str) -> str:
 def print_results(result: Result, batch_size: int = 20) -> None:
     """Print query results in a paginated table format."""
     try:
-        if result.returns_rows:  # type: ignore[attr-defined]
+        if result.returns_rows:  # type: ignore
             rows = result.fetchall()
             if not rows:
                 print("Query returned no results.")