dev: select columns instead of creating

lucas-nelson-uiuc · Dec 26, 2024 · b79c63d · b79c63d
1 parent c415087
commit b79c63d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/tidy_tools/model/model.py b/src/tidy_tools/model/model.py
@@ -130,9 +130,10 @@ def convert(cls, data: DataFrame):
             for cls_field in attrs.fields(cls)
         }
 
-        return data.withColumns(
-            {cls_field.name: column for cls_field, column in queue.items()}
-        )
+        # return data.withColumns(
+        #     {cls_field.name: column for cls_field, column in queue.items()}
+        # )
+        return data.select(*(column for column in queue.values()))
 
     @classmethod
     def validate(cls, data: DataFrame) -> DataFrame: