From e53e179cf11dada93b0522517b28bc8d4491505e Mon Sep 17 00:00:00 2001 From: Wilson Wang Date: Wed, 2 Nov 2022 10:56:54 -0700 Subject: [PATCH] bug fix --- benchmark/basic.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/benchmark/basic.py b/benchmark/basic.py index 903a2e5..c40caaf 100644 --- a/benchmark/basic.py +++ b/benchmark/basic.py @@ -16,18 +16,18 @@ # import ray -import apache_beam.tools as tools +from apache_beam.tools import fn_api_runner_microbenchmark as bm def run_pipeline(): - tools.run_benchmark(1, 20, 100, True) + bm.run_benchmark(1, 20, 100, True) return None def test_basic_benchmark(benchmark): ray.init() - res = benchmark(tools.run_pipeline) + res = benchmark(run_pipeline) ray.shutdown() assert res is None