Merge pull request apache#2 from chamikaramj/labels

Fixes examples
cosmoskitten · Jul 22, 2016 · cc5feaa · cc5feaa
2 parents df3511d + 43aafcb
commit cc5feaa
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 9 deletions.
diff --git a/sdks/python/apache_beam/examples/complete/autocomplete.py b/sdks/python/apache_beam/examples/complete/autocomplete.py
@@ -51,8 +51,8 @@ def run(argv=None):
 
 class TopPerPrefix(beam.PTransform):
 
-  def __init__(self, label, count):
-    super(TopPerPrefix, self).__init__(label)
+  def __init__(self, count):
+    super(TopPerPrefix, self).__init__()
     self._count = count
 
   def apply(self, words):

diff --git a/sdks/python/apache_beam/examples/complete/estimate_pi.py b/sdks/python/apache_beam/examples/complete/estimate_pi.py
@@ -88,9 +88,6 @@ def encode(self, x):
 class EstimatePiTransform(beam.PTransform):
   """Runs 10M trials, and combine the results to estimate pi."""
 
-  def __init__(self, label):
-    super(EstimatePiTransform, self).__init__(label)
-
   def apply(self, pcoll):
     # A hundred work items of a hundred thousand tries each.
     return (pcoll

diff --git a/sdks/python/apache_beam/examples/snippets/snippets.py b/sdks/python/apache_beam/examples/snippets/snippets.py
@@ -466,7 +466,7 @@ def examples_wordcount_minimal(renames):
       # [END examples_wordcount_minimal_map]
 
       # [START examples_wordcount_minimal_write]
-      | 'gs://my-bucket/counts.txt' >> beam.io.Write(beam.io.TextFileSink())
+      | beam.io.Write(beam.io.TextFileSink('gs://my-bucket/counts.txt'))
       # [END examples_wordcount_minimal_write]
   )
 
@@ -531,7 +531,7 @@ def process(self, context):
   formatted = counts | beam.ParDo(FormatAsTextFn())
   # [END examples_wordcount_wordcount_dofn]
 
-  formatted | 'gs://my-bucket/counts.txt' >> beam.io.Write(beam.io.TextFileSink())
+  formatted |  beam.io.Write(beam.io.TextFileSink('gs://my-bucket/counts.txt'))
   p.visit(SnippetUtils.RenameFiles(renames))
   p.run()
 
@@ -702,8 +702,8 @@ class _CountingSource(CountingSource):
   # [START model_custom_source_new_ptransform]
   class ReadFromCountingSource(PTransform):
 
-    def __init__(self, label, count, **kwargs):
-      super(ReadFromCountingSource, self).__init__(label, **kwargs)
+    def __init__(self, count, **kwargs):
+      super(ReadFromCountingSource, self).__init__(**kwargs)
       self._count = count
 
     def apply(self, pcoll):