Skip to content

Commit

Permalink
Merge pull request #4241 from pabloem/patch-2
Browse files Browse the repository at this point in the history
[BEAM-3042] Renaming properties form IO target counter name.
  • Loading branch information
aaltay committed Dec 12, 2017
2 parents f7fc4bd + b7396c4 commit b97df71
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions sdks/python/apache_beam/utils/counters.py
Expand Up @@ -29,19 +29,18 @@
from apache_beam.transforms import cy_combiners

# Information identifying the IO being measured by a counter.
IOTargetName = namedtuple('IOTargetName', ['side_input_step_name',
'side_input_index',
'original_shuffle_step_name'])
IOTargetName = namedtuple('IOTargetName', ['requesting_step_name',
'input_index'])


def side_input_id(step_name, input_index):
"""Create an IOTargetName that identifies the reading of a side input."""
return IOTargetName(step_name, input_index, None)
return IOTargetName(step_name, input_index)


def shuffle_id(step_name):
"""Create an IOTargetName that identifies a GBK step."""
return IOTargetName(None, None, step_name)
return IOTargetName(step_name, None)


_CounterName = namedtuple('_CounterName', ['name',
Expand Down

0 comments on commit b97df71

Please sign in to comment.