Add stubs and improve comments for pyo3-exposed classes, driveby type…

… fixes.
Eventual-Inc · Sep 14, 2023 · bdb4544 · bdb4544
1 parent fc10d6d
commit bdb4544
Show file tree

Hide file tree

Showing 21 changed files with 714 additions and 86 deletions.
diff --git a/daft/context.py b/daft/context.py
@@ -109,7 +109,10 @@ def logical_plan_builder_class(self) -> type[LogicalPlanBuilder]:
         from daft.logical.logical_plan import PyLogicalPlanBuilder
         from daft.logical.rust_logical_plan import RustLogicalPlanBuilder
 
-        return RustLogicalPlanBuilder if self.use_rust_planner else PyLogicalPlanBuilder
+        if self.use_rust_planner:
+            return RustLogicalPlanBuilder
+        else:
+            return PyLogicalPlanBuilder
 
 
 _DaftContext = DaftContext()