append content type params if intelligent estimation is enabled

ray-project · Sep 25, 2024 · b5d0d05 · b5d0d05
1 parent edff05d
commit b5d0d05
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/deltacat/compute/compactor_v2/utils/io.py b/deltacat/compute/compactor_v2/utils/io.py
@@ -109,7 +109,7 @@ def create_uniform_input_deltas(
     input_da_list = []
 
     for delta in input_deltas:
-        if enable_input_split:
+        if enable_input_split or enable_intelligent_size_estimation:
             # An idempotent operation to ensure content type params exist
             append_content_type_params(
                 delta=delta,

diff --git a/deltacat/tests/compute/test_util_common.py b/deltacat/tests/compute/test_util_common.py
@@ -267,6 +267,8 @@ def assert_compaction_audit(
             compaction_audit.peak_memory_used_bytes_per_task,
             compaction_audit.pyarrow_version,
             compaction_audit.telemetry_time_in_seconds,
+            compaction_audit.observed_input_inflation,
+            compaction_audit.observed_input_average_record_size_bytes,
         ]
         for entry in audit_entries:
             assert entry is not None