Fix SessionContext execute. (#353)

apache · Apr 27, 2023 · 6642660 · 6642660
1 parent e1a8a3c
commit 6642660
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/src/context.rs b/src/context.rs
@@ -712,14 +712,7 @@ impl PySessionContext {
         part: usize,
         py: Python,
     ) -> PyResult<PyRecordBatchStream> {
-        let ctx = TaskContext::new(
-            None,
-            "session_id".to_string(),
-            SessionConfig::new(),
-            HashMap::new(),
-            HashMap::new(),
-            Arc::new(RuntimeEnv::default()),
-        );
+        let ctx: TaskContext = TaskContext::from(&self.ctx.state());
         // create a Tokio runtime to run the async code
         let rt = &get_tokio_runtime(py).0;
         let plan = plan.plan.clone();