address comments

Author: Yingjian Wu

pyiceberg/table/snapshots.py

@@ -277,6 +277,9 @@ def __init__(self, partition_summary_limit: int = 0) -> None:
         self.partition_metrics = defaultdict(UpdateMetrics)
         self.max_changed_partitions_for_summaries = partition_summary_limit
 
+    def set_partition_summary_limit(self, limit: int) -> None:
+        self.max_changed_partitions_for_summaries = limit
+
     def add_file(self, data_file: DataFile, schema: Schema, partition_spec: PartitionSpec = UNPARTITIONED_PARTITION_SPEC) -> None:
         self.metrics.add_file(data_file)
         if len(data_file.partition) > 0:

tests/table/test_snapshots.py

@@ -201,6 +201,22 @@ def test_snapshot_summary_collector_with_partition() -> None:
         "changed-partition-count": "2",
     }
 
+    # When
+    ssc.set_partition_summary_limit(10)
+
+    # Then
+    assert ssc.build() == {
+        "added-files-size": "1234",
+        "removed-files-size": "5555",
+        "added-data-files": "1",
+        "deleted-data-files": "2",
+        "added-records": "100",
+        "deleted-records": "300",
+        "changed-partition-count": "2",
+        "partitions.int_field=1": "added-files-size=1234,removed-files-size=1234,added-data-files=1,deleted-data-files=1,added-records=100,deleted-records=100",
+        "partitions.int_field=2": "removed-files-size=4321,deleted-data-files=1,deleted-records=200",
+    }
+
 
 @pytest.mark.integration
 def test_snapshot_summary_collector_with_partition_limit_in_constructor() -> None: