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

pabloem pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git

commit 22f5a81e49996676f10200065ca3d0afeb0bb1a9
Author: Pablo <[email protected]>
AuthorDate: Tue May 8 17:15:36 2018 -0700

    Adding the sampling thread to side input microbench.
---
 sdks/python/apache_beam/tools/sideinput_microbenchmark.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/sdks/python/apache_beam/tools/sideinput_microbenchmark.py 
b/sdks/python/apache_beam/tools/sideinput_microbenchmark.py
index 7cfdb29..f517304 100644
--- a/sdks/python/apache_beam/tools/sideinput_microbenchmark.py
+++ b/sdks/python/apache_beam/tools/sideinput_microbenchmark.py
@@ -51,6 +51,7 @@ def run_benchmark(num_runs=50, input_per_source=4000, 
num_sources=4):
   for i in range(num_runs):
     counter_factory = CounterFactory()
     state_sampler = statesampler.StateSampler('basic', counter_factory)
+    state_sampler.start()
     with state_sampler.scoped_state('step1', 'state'):
       si_counter = opcounters.SideInputReadCounter(
           counter_factory, state_sampler, 'step1', 1)
@@ -64,6 +65,7 @@ def run_benchmark(num_runs=50, input_per_source=4000, 
num_sources=4):
       list(iterator_fn())
       time_cost = time.time() - start
       times.append(time_cost)
+    state_sampler.stop()
 
   print("Runtimes:", times)
 

-- 
To stop receiving notification emails like this one, please contact
[email protected].

Reply via email to