Merge pull request apache#10122: Fixes a failing check.

mransley · Nov 15, 2019 · 2f00613 · 2f00613
2 parents b653c29 + 310f289
commit 2f00613
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/sdks/python/apache_beam/transforms/core.py b/sdks/python/apache_beam/transforms/core.py
@@ -379,16 +379,16 @@ def to_runner_api(self, context, has_parts=False):
  else:
  env1 = id_to_proto_map[env_id]
  env2 = context.environments[env_id]
- assert env1.urn == env2.proto.urn, (
+ assert env1.urn == env2.to_runner_api(context).urn, (
  'Expected environments with the same ID to be equal but received '
  'environments with different URNs '
  '%r and %r',
- env1.urn, env2.proto.urn)
- assert env1.payload == env2.proto.payload, (
+ env1.urn, env2.to_runner_api(context).urn)
+ assert env1.payload == env2.to_runner_api(context).payload, (
  'Expected environments with the same ID to be equal but received '
  'environments with different payloads '
  '%r and %r',
- env1.payload, env2.proto.payload)
+ env1.payload, env2.to_runner_api(context).payload)
  return self._proto
 
  def get_restriction_coder(self):