bug fix

ray-project · Nov 2, 2022 · e53e179 · e53e179
1 parent 342845e
commit e53e179
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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