Skip to content

Commit 6d6a64f

Browse files
committed
delta_lake_loader: Simplify batch statistics, no compaction stats
1 parent 51c125b commit 6d6a64f

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

src/amp/loaders/implementations/deltalake_loader.py

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -201,11 +201,7 @@ def _load_batch_impl(self, batch: pa.RecordBatch, table_name: str, **kwargs) ->
201201
self._refresh_table_reference()
202202

203203
# Post-write optimizations
204-
optimization_results = self._perform_post_write_optimizations(table.num_rows)
205-
206-
# Store optimization results in base class metadata
207-
if hasattr(self, '_last_batch_metadata'):
208-
self._last_batch_metadata = optimization_results
204+
_optimization_results = self._perform_post_write_optimizations()
209205

210206
return batch.num_rows
211207

@@ -281,7 +277,7 @@ def _refresh_table_reference(self) -> None:
281277
self.logger.error(f'Failed to refresh table reference: {e}')
282278
# Don't set _table_exists = False here as the table might still exist
283279

284-
def _perform_post_write_optimizations(self, rows_written: int) -> Dict[str, Any]:
280+
def _perform_post_write_optimizations(self) -> Dict[str, Any]:
285281
"""Perform post-write optimizations with robust API handling"""
286282
optimization_results = {}
287283

@@ -453,10 +449,6 @@ def _get_loader_batch_metadata(self, batch: pa.RecordBatch, duration: float, **k
453449
if self._table_exists and self._delta_table is not None:
454450
metadata['table_version'] = self._delta_table.version()
455451

456-
# Add optimization results if available
457-
if hasattr(self, '_last_batch_metadata'):
458-
metadata['optimization_results'] = self._last_batch_metadata
459-
460452
return metadata
461453

462454
def _get_loader_table_metadata(

0 commit comments

Comments
 (0)