nehsyc commented on a change in pull request #13474:
URL: https://github.com/apache/beam/pull/13474#discussion_r535706100
##########
File path: sdks/python/apache_beam/coders/coders.py
##########
@@ -1485,6 +1486,21 @@ def as_cloud_object(self, coders_context=None):
],
}
+ def to_type_hint(self):
+ from apache_beam.typehints import sharded_key_type
+ return sharded_key_type.ShardedKeyTypeConstraint(
+ self._key_coder.to_type_hint())
+
+ @staticmethod
+ def from_type_hint(typehint, registry):
+ from apache_beam.typehints import sharded_key_type
+ if isinstance(typehint, sharded_key_type.ShardedKeyTypeConstraint):
+ return ShardedKeyCoder(registry.get_coder(typehint.key_type))
+ else:
+ raise ValueError((
Review comment:
Done.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
