Small fix

Eventual-Inc · May 1, 2024 · 9aac19f · 9aac19f
1 parent 60da15f
commit 9aac19f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/daft-scan/src/lib.rs b/src/daft-scan/src/lib.rs
@@ -504,9 +504,9 @@ impl ScanTask {
                             config.csv_inflation_factor
                         }
                         #[cfg(feature = "python")]
-                        FileFormatConfig::Database(_) => 0.0,
+                        FileFormatConfig::Database(_) => 1.0,
                         #[cfg(feature = "python")]
-                        FileFormatConfig::PythonFunction => 0.0,
+                        FileFormatConfig::PythonFunction => 1.0,
                     };
 
                     // estimate number of rows from read schema

diff --git a/tests/io/lancedb/test_lancedb_reads.py b/tests/io/lancedb/test_lancedb_reads.py
@@ -41,3 +41,10 @@ def test_lancedb_read_filter(lance_dataset_path):
     df = df.where(df["lat"] > 45)
     df = df.select("vector")
     assert df.to_pydict() == {"vector": data["vector"][:1]}
+
+
+def test_lancedb_read_limit(lance_dataset_path):
+    df = daft.read_lance(lance_dataset_path)
+    df = df.limit(1)
+    df = df.select("vector")
+    assert df.to_pydict() == {"vector": data["vector"][:1]}