Skip to content

Commit

Permalink
[BUG] [Hotfix] Fix limit pushdown test. (#1728)
Browse files Browse the repository at this point in the history
This PR fixes an uncaught compilation error in a test that's not run (or
compiled) in CI.
  • Loading branch information
clarkzinzow authored Dec 15, 2023
1 parent 52e4809 commit b833b25
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions src/daft-plan/src/optimization/rules/push_down_limit.rs
Original file line number Diff line number Diff line change
Expand Up @@ -195,9 +195,10 @@ mod tests {

let py_obj = Python::with_gil(|py| py.None());
let schema: Arc<Schema> = Schema::new(vec![Field::new("a", DataType::Int64)])?.into();
let plan = LogicalPlanBuilder::in_memory_scan("foo", py_obj, schema, Default::default())?
.limit(5, false)?
.build();
let plan =
LogicalPlanBuilder::in_memory_scan("foo", py_obj, schema, Default::default(), 5)?
.limit(5, false)?
.build();
let expected = "\
Limit: 5\
\n . Source: Json, File paths = [/foo], File schema = a (Int64), b (Utf8), Format-specific config = Json(JsonSourceConfig { buffer_size: None, chunk_size: None }), Storage config = Native(NativeStorageConfig { io_config: None, multithreaded_io: true }), Output schema = a (Int64), b (Utf8)";
Expand Down

0 comments on commit b833b25

Please sign in to comment.