diff --git a/daft/dataframe/dataframe.py b/daft/dataframe/dataframe.py index 78efbeb818..edf787ddc8 100644 --- a/daft/dataframe/dataframe.py +++ b/daft/dataframe/dataframe.py @@ -629,6 +629,7 @@ def _from_schema(cls, schema: Schema) -> "DataFrame": # Write methods ### + @BroadcastMetrics @DataframePublicAPI def write_parquet( self, @@ -708,6 +709,7 @@ def write_parquet( } ) + @BroadcastMetrics @DataframePublicAPI def write_csv( self, @@ -779,6 +781,7 @@ def write_csv( } ) + @BroadcastMetrics @DataframePublicAPI def write_iceberg( self, table: "pyiceberg.table.Table", mode: str = "append", io_config: Optional[IOConfig] = None @@ -929,6 +932,7 @@ def write_iceberg( # This is due to the fact that the logical plan of the write_iceberg returns datafiles but we want to return the above data return from_pydict(with_operations) + @BroadcastMetrics @DataframePublicAPI def write_deltalake( self, @@ -1140,6 +1144,7 @@ def write_deltalake( return with_operations + @BroadcastMetrics @DataframePublicAPI def write_lance( self,