Move buffered_scheduler out of local (#157)

mtrofin · web-flow · commit 12c239b8b8c4 · 2022-09-29T14:39:03.000-07:00
It's got nothing to do with "local", it is reusable just fine as-is in
distributed scenarios.
diff --git a/compiler_opt/distributed/buffered_scheduler.py b/compiler_opt/distributed/buffered_scheduler.py
diff --git a/compiler_opt/distributed/buffered_scheduler_test.py b/compiler_opt/distributed/buffered_scheduler_test.py
@@ -20,7 +20,7 @@
 
 from absl.testing import absltest
 from compiler_opt.distributed import worker
-from compiler_opt.distributed.local import buffered_scheduler
+from compiler_opt.distributed import buffered_scheduler
 
 
 class BufferedSchedulerTest(absltest.TestCase):
diff --git a/compiler_opt/rl/local_data_collector.py b/compiler_opt/rl/local_data_collector.py
@@ -23,7 +23,7 @@
 from tf_agents.trajectories import trajectory
 
 from compiler_opt.distributed import worker
-from compiler_opt.distributed.local import buffered_scheduler
+from compiler_opt.distributed import buffered_scheduler
 from compiler_opt.rl import compilation_runner
 from compiler_opt.rl import corpus
 from compiler_opt.rl import data_collector
