[CHORE] Add schemas for remaining local plan ops (#3446)

Co-authored-by: Colin Ho <[email protected]>
Eventual-Inc · Nov 28, 2024 · 151a2cf · 151a2cf
1 parent 0e7333a
commit 151a2cf
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/daft-local-plan/src/plan.rs b/src/daft-local-plan/src/plan.rs
@@ -373,9 +373,14 @@ impl LocalPhysicalPlan {
             | Self::HashJoin(HashJoin { schema, .. })
             | Self::Explode(Explode { schema, .. })
             | Self::Unpivot(Unpivot { schema, .. })
-            | Self::Concat(Concat { schema, .. }) => schema,
+            | Self::Concat(Concat { schema, .. })
+            | Self::MonotonicallyIncreasingId(MonotonicallyIncreasingId { schema, .. }) => schema,
+            Self::PhysicalWrite(PhysicalWrite { file_schema, .. }) => file_schema,
             Self::InMemoryScan(InMemoryScan { info, .. }) => &info.source_schema,
-            _ => todo!("{:?}", self),
+            #[cfg(feature = "python")]
+            Self::CatalogWrite(CatalogWrite { file_schema, .. }) => file_schema,
+            #[cfg(feature = "python")]
+            Self::LanceWrite(LanceWrite { file_schema, .. }) => file_schema,
         }
     }
 }