diff --git a/keras_core/utils/io_utils.py b/keras_core/utils/io_utils.py index f47b8c29f..21127b32e 100644 --- a/keras_core/utils/io_utils.py +++ b/keras_core/utils/io_utils.py @@ -6,7 +6,12 @@ from keras_core.api_export import keras_core_export from keras_core.backend.common import global_state -@keras_core_export("keras_core.config.enable_interactive_logging") +@keras_core_export( + [ + "keras_core.config.enable_interactive_logging", + "keras_core.utils.enable_interactive_logging", + ] +) def enable_interactive_logging(): """Turn on interactive logging. @@ -17,7 +22,12 @@ def enable_interactive_logging(): global_state.set_global_setting("interactive_logging", True) -@keras_core_export("keras_core.config.disable_interactive_logging") +@keras_core_export( + [ + "keras_core.config.disable_interactive_logging", + "keras_core.utils.disable_interactive_logging", + ] +) def disable_interactive_logging(): """Turn off interactive logging. @@ -28,7 +38,12 @@ def disable_interactive_logging(): global_state.set_global_setting("interactive_logging", False) -@keras_core_export("keras_core.config.is_interactive_logging_enabled") +@keras_core_export( + [ + "keras_core.config.is_interactive_logging_enabled", + "keras_core.utils.is_interactive_logging_enabled", + ] +) def is_interactive_logging_enabled(): """Check if interactive logging is enabled.