Skip to content

Commit 7c4076c

Browse files
dorellangcopybara-github
authored andcommitted
Allow passing ICEBERG as an optional component for Dataproc.
PiperOrigin-RevId: 842887104
1 parent ca58bbb commit 7c4076c

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

perfkitbenchmarker/configs/benchmark_config_spec.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,12 @@ class _DpbApplicationListDecoder(option_decoders.ListDecoder):
5454
def __init__(self, **kwargs):
5555
super().__init__(
5656
default=None,
57-
item_decoder=option_decoders.EnumDecoder(
58-
[dpb_constants.FLINK, dpb_constants.HIVE, dpb_constants.DELTA]
59-
),
57+
item_decoder=option_decoders.EnumDecoder([
58+
dpb_constants.FLINK,
59+
dpb_constants.HIVE,
60+
dpb_constants.DELTA,
61+
dpb_constants.ICEBERG,
62+
]),
6063
**kwargs,
6164
)
6265

perfkitbenchmarker/dpb_constants.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@
6666
FLINK = 'flink'
6767
HIVE = 'hive'
6868
DELTA = 'DELTA'
69+
ICEBERG = 'ICEBERG'
6970

7071
# Metrics and Status related metadata
7172
# TODO(pclay): Remove these after migrating all callers to SubmitJob

0 commit comments

Comments
 (0)