renamed log_handlers_builder to handler_factory

Author: apoorvkh

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "torchrunx"
-version = "0.2.1"
+version = "0.2.2"
 authors = [
     {name = "Apoorv Khandelwal", email = "[email protected]"},
     {name = "Peter Curtin", email = "[email protected]"},

src/torchrunx/launcher.py

@@ -64,7 +64,7 @@ def run(  # noqa: C901, PLR0912
         func: Callable,
         func_args: tuple[Any] | None = None,
         func_kwargs: dict[str, Any] | None = None,
-        log_handlers_builder: Callable[[], list[Handler]] | Literal["auto"] | None = "auto",
+        handler_factory: Callable[[], list[Handler]] | Literal["auto"] | None = "auto",
     ) -> LaunchResult:
         """Run a function using the :mod:`torchrunx.Launcher` configuration."""
         if not dist.is_available():
@@ -88,7 +88,7 @@ def run(  # noqa: C901, PLR0912
             # Start logging server (recieves LogRecords from agents/workers)
 
             logging_server_args = LoggingServerArgs(
-                log_handlers_builder=log_handlers_builder,
+                handler_factory=handler_factory,
                 logging_hostname=launcher_hostname,
                 logging_port=logging_port,
                 hostnames=hostnames,
@@ -214,7 +214,7 @@ def launch(
     ),
     extra_env_vars: tuple[str, ...] = (),
     env_file: str | os.PathLike | None = None,
-    log_handlers_builder: Callable[[], list[Handler]] | Literal["auto"] | None = "auto",
+    handler_factory: Callable[[], list[Handler]] | Literal["auto"] | None = "auto",
 ) -> LaunchResult:
     """Launch a distributed PyTorch function on the specified nodes.
 
@@ -235,7 +235,7 @@ def launch(
             Supports bash pattern matching syntax.
         extra_env_vars: Additional user-specified environment variables to copy.
         env_file: Path to a file (e.g., `.env`) with additional environment variables to copy.
-        log_handlers_builder: Function to build handlers for managing agent and worker logs.
+        handler_factory: Function to build logging handlers that process agent and worker logs.
             Defaults to an automatic basic logging scheme.
 
     Raises:
@@ -257,7 +257,7 @@ def launch(
         func=func,
         func_args=func_args,
         func_kwargs=func_kwargs,
-        log_handlers_builder=log_handlers_builder,
+        handler_factory=handler_factory,
     )
 
 

src/torchrunx/utils/logging.py

@@ -187,7 +187,7 @@ def shutdown(self) -> None:
 class LoggingServerArgs:
     """Arguments for starting a :class:`_LogRecordSocketReceiver`."""
 
-    log_handlers_builder: Callable[[], list[Handler]] | Literal["auto"] | None
+    handler_factory: Callable[[], list[Handler]] | Literal["auto"] | None
     logging_hostname: str
     logging_port: int
     hostnames: list[str]
@@ -213,17 +213,17 @@ def start_logging_server(
     args = LoggingServerArgs.deserialize(serialized_args)
 
     log_handlers = []
-    if args.log_handlers_builder is None:
+    if args.handler_factory is None:
         log_handlers = []
-    elif args.log_handlers_builder == "auto":
+    elif args.handler_factory == "auto":
         log_handlers = default_handlers(
             hostnames=args.hostnames,
             workers_per_host=args.workers_per_host,
             log_dir=args.log_dir,
             log_level=args.log_level,
         )
-    elif isinstance(args.log_handlers_builder, Callable):
-        log_handlers = args.log_handlers_builder()
+    elif isinstance(args.handler_factory, Callable):
+        log_handlers = args.handler_factory()
 
     log_receiver = _LogRecordSocketReceiver(
         host=args.logging_hostname,