tvalentyn commented on a change in pull request #14390:
URL: https://github.com/apache/beam/pull/14390#discussion_r616918063
##########
File path: sdks/python/apache_beam/transforms/ptransform.py
##########
@@ -422,6 +423,35 @@ def with_output_types(self, type_hint):
validate_composite_type_param(type_hint, 'Type hints for a PTransform')
return super(PTransform, self).with_output_types(type_hint)
+ def with_resource_hints(self, **kwargs): # type: (...) -> PTransform
+ """Adds resource hints to the :class:`PTransform`.
+
+ Resource hints allow users to express constraints on the environment where
+ the transform should be executed. Interpretation of the resource hints is
+ defined by Beam Runners. Runners may ignore unsupported the hints.
+
+ Args:
+ **kwargs: key-value pairs describing hints and their values.
+
+ Raises:
+ ValueError: if provided hints are unknown to the SDK. See
+ :mod:~apache_beam.transforms.resources` for a list of known hints.
+
+ Returns:
+ PTransform: A reference to the instance of this particular
+ :class:`PTransform` object.
+ """
+ self._resource_hints = resources.parse_resource_hints(kwargs)
Review comment:
Good point, thanks. Fixed.
--
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]
