[beam] 01/01: Add output typehints to GroupIntoBatches

[al...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

altay pushed a commit to branch aaltay-patch-2
in repository https://gitbox.apache.org/repos/asf/beam.git

commit 91aa2269e33ed5478277c786a1db74cc16432046
Author: Ahmet Altay <aa...@gmail.com>
AuthorDate: Wed Jun 24 19:05:19 2020 -0700

    Add output typehints to GroupIntoBatches
---
 sdks/python/apache_beam/transforms/util.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/sdks/python/apache_beam/transforms/util.py b/sdks/python/apache_beam/transforms/util.py
index afb4192..89704e1 100644
--- a/sdks/python/apache_beam/transforms/util.py
+++ b/sdks/python/apache_beam/transforms/util.py
@@ -741,6 +741,7 @@ def WithKeys(pcoll, k):
 
 @experimental()
 @typehints.with_input_types(Tuple[K, V])
+@typehints.with_output_types(Tuple[K, List[V]])
 class GroupIntoBatches(PTransform):
   """PTransform that batches the input into desired batch size. Elements are
   buffered until they are equal to batch size provided in the argument at which