changed name for utils.logging (ambiguous)

Author: apoorvkh

pyproject.toml

@@ -48,8 +48,6 @@ src = ["src", "tests"]
 [tool.ruff.lint]
 select = ["ALL"]
 ignore = [
-  "ANN101",
-  "ANN102",
   "ANN401",  # self / cls / Any annotations
   "BLE001",  # blind exceptions
   "TD",      # todo syntax

src/torchrunx/__init__.py

@@ -2,7 +2,7 @@
 
 from .launcher import Launcher, LaunchResult, launch
 from .utils.errors import AgentFailedError, WorkerFailedError
-from .utils.logging import add_filter_to_handler, file_handler, stream_handler
+from .utils.logging_server import add_filter_to_handler, file_handler, stream_handler
 
 __all__ = [
     "AgentFailedError",

src/torchrunx/agent.py

@@ -19,7 +19,7 @@
     LauncherAgentGroup,
     get_open_port,
 )
-from .utils.logging import log_records_to_socket, redirect_stdio_to_logger
+from .utils.logging_server import log_records_to_socket, redirect_stdio_to_logger
 from .worker import WorkerArgs, worker_entrypoint
 
 

src/torchrunx/launcher.py

@@ -34,7 +34,7 @@
     ExceptionFromWorker,
     WorkerFailedError,
 )
-from .utils.logging import LoggingServerArgs, start_logging_server
+from .utils.logging_server import LoggingServerArgs, start_logging_server
 
 
 @dataclass

src/torchrunx/utils/logging.py renamed to src/torchrunx/utils/logging_server.py

@@ -15,7 +15,7 @@
 import torch.distributed as dist
 
 from .utils.errors import ExceptionFromWorker
-from .utils.logging import log_records_to_socket, redirect_stdio_to_logger
+from .utils.logging_server import log_records_to_socket, redirect_stdio_to_logger
 
 __all__ = ["WorkerArgs", "worker_entrypoint"]