new unchecked table

Eventual-Inc · Dec 12, 2023 · 48d039b · 48d039b
1 parent 7911834
commit 48d039b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/daft-csv/src/read.rs b/src/daft-csv/src/read.rs
@@ -289,7 +289,7 @@ async fn read_csv_single_into_table(
             .zip(&read_fields)
             .map(|(array, field)| Series::try_from((field.name.as_ref(), array)))
             .collect::<DaftResult<Vec<Series>>>()?;
-        let table = Table::new(owned_read_schema.clone(), columns_series)?;
+        let table = Table::new_unchecked(owned_read_schema.clone(), columns_series);
         if let Some(predicate) = &predicate {
             let filtered = table.filter(&[predicate.as_ref()])?;
             if let Some(include_columns) = &include_columns {

diff --git a/src/daft-table/src/lib.rs b/src/daft-table/src/lib.rs
@@ -71,6 +71,14 @@ impl Table {
         })
     }
 
+
+    pub fn new_unchecked<S: Into<SchemaRef>>(schema: S, columns: Vec<Series>) -> Self {
+        Table {
+            schema: schema.into(),
+            columns: columns,
+        }
+    }
+
     pub fn empty(schema: Option<SchemaRef>) -> DaftResult<Self> {
         match schema {
             Some(schema) => {