Skip to content

Commit 487f1e8

Browse files
committed
clean up dist accesses
1 parent 24d92fe commit 487f1e8

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

sdks/python/apache_beam/runners/dataflow/dataflow_metrics.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -210,10 +210,10 @@ def _get_metric_value(self, metric):
210210
return metric.scalar
211211
elif metric.distribution is not None:
212212
print("Treating as distribution: ", metric.distribution)
213-
dist_count = metric.distribution.struct_value.fields['count'].number_value
214-
dist_min = metric.distribution.struct_value.fields['min'].number_value
215-
dist_max = metric.distribution.struct_value.fields['max'].number_value
216-
dist_sum = metric.distribution.struct_value.fields['sum'].number_value
213+
dist_count = metric.distribution['count']
214+
dist_min = metric.distribution['min']
215+
dist_max = metric.distribution['max']
216+
dist_sum = metric.distribution['sum']
217217
return DistributionResult(
218218
DistributionData(dist_sum, dist_count, dist_min, dist_max))
219219
#TODO(https://github.com/apache/beam/issues/31788) support StringSet after

0 commit comments

Comments
 (0)