collect into a single batches

Lordworms · Feb 14, 2024 · 455a5d9 · 455a5d9
1 parent e13670e
commit 455a5d9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/datafusion/core/src/execution/context/mod.rs b/datafusion/core/src/execution/context/mod.rs
@@ -947,8 +947,7 @@ impl SessionContext {
  } else {
  Arc::new(Schema::empty())
  };
- let provider =
- MemTable::try_new(schema, batches.map(|batch| vec![batch]).collect())?;
+ let provider = MemTable::try_new(schema, vec![batches.collect()])?;
  Ok(DataFrame::new(
  self.state(),
  LogicalPlanBuilder::scan(