migrate existing set_logging_level callsites

Summary:
**Diff Purpose & Changes**
Same as D72941480, but for set_worker_logging_level

Reviewed By: colin2328

Differential Revision: D72942837

fbshipit-source-id: 83115a6915ec353fb295f72baf6285dcc45fa24a

Author: amirafzali

python/monarch/worker/worker.py

@@ -54,10 +54,6 @@
     CONTROLLER_COMPILED_REPEAT = True
 
 
-def set_worker_logging_level(level):
-    logger.setLevel(level)
-
-
 def set_random_seed_impl(seed, process_idx):
     torch.manual_seed(seed ^ process_idx)
 

python/tests/dispatch_bench.py

@@ -7,13 +7,13 @@
 
 # this function helps get a local device mesh for testing
 from monarch._testing import mock_mesh
+from monarch.builtins.log import set_logging_level_remote
 
 from monarch.common._coalescing import coalescing
 from monarch.common.remote import remote
 from monarch.fetch import fetch_shard
 from monarch.python_local_mesh import python_local_mesh
 from monarch_supervisor.logging import initialize_logging
-
 from tests.dispatch_bench_helper import run_loop, run_loop_local
 
 NITER = 10000
@@ -25,10 +25,6 @@
 # user-defined remote functions
 log = remote("monarch.worker._testing_function.log", propagate="inspect")
 
-set_worker_logging_level = remote(
-    "monarch.worker.worker.set_worker_logging_level", propagate="inspect"
-)
-
 
 def local_run():
     run_loop_local(NITER, DEFAULT_TENSOR_SIZE)
@@ -66,7 +62,7 @@ def main():
 
     with device_mesh.activate():
         torch.set_default_device("cuda")
-        set_worker_logging_level(logging.WARNING)
+        set_logging_level_remote(logging.WARNING)
 
     # bench 1: local compute only
     t0 = benchmark.Timer(