From ef77887e1c624a965039a3a0fcd443db1fdd5973 Mon Sep 17 00:00:00 2001 From: Luke Baumann Date: Wed, 13 May 2026 10:51:36 -0700 Subject: [PATCH] Ensure all logger instances in pathwaysutils are _logger PiperOrigin-RevId: 914948052 --- pathwaysutils/persistence/orbax_handler.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pathwaysutils/persistence/orbax_handler.py b/pathwaysutils/persistence/orbax_handler.py index 0a992b9..7417d12 100644 --- a/pathwaysutils/persistence/orbax_handler.py +++ b/pathwaysutils/persistence/orbax_handler.py @@ -29,7 +29,7 @@ from pathwaysutils.persistence import helper -logger = logging.getLogger(__name__) +_logger = logging.getLogger(__name__) ParamInfo = type_handlers.ParamInfo SaveArgs = type_handlers.SaveArgs @@ -197,7 +197,7 @@ async def deserialize( mesh_axes.append(sharding.spec) shardings.append(sharding) if arg.global_shape is None or arg.dtype is None: - logger.warning( + _logger.warning( "Shape or dtype not provided for restoration. Provide these" " properties for improved performance." ) @@ -279,7 +279,7 @@ def register_pathways_handlers( array_metadata_store: array_metadata_store_lib.Store | None = None, ): """Function that must be called before saving or restoring with Pathways.""" - logger.debug( + _logger.debug( "Registering CloudPathwaysArrayHandler (Pathways Persistence API)." ) type_handlers.register_type_handler(