@ -13,7 +13,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
# BMT ools copied from Huggingface Transformers# swarms.t ools copied from Huggingface Transformers
""" Logging utilities. """ """ Logging utilities. """
import logging import logging
@ -47,16 +47,16 @@ _default_log_level = logging.INFO
def _get_default_logging_level ( ) : def _get_default_logging_level ( ) :
"""
"""
If BM TOOLS_VERBOSITY env var is set to one of the valid choices return that as the new default level . If it is
If SWARMS TOOLS_VERBOSITY env var is set to one of the valid choices return that as the new default level . If it is
not - fall back to ` ` _default_log_level ` `
not - fall back to ` ` _default_log_level ` `
"""
"""
env_level_str = os . getenv ( " BM TOOLS_VERBOSITY" , None )
env_level_str = os . getenv ( " SWARMS TOOLS_VERBOSITY" , None )
if env_level_str :
if env_level_str :
if env_level_str in log_levels :
if env_level_str in log_levels :
return log_levels [ env_level_str ]
return log_levels [ env_level_str ]
else :
else :
logging . getLogger ( ) . warning (
logging . getLogger ( ) . warning (
f " Unknown option BM TOOLS_VERBOSITY={ env_level_str } , "
f " Unknown option SWARMS TOOLS_VERBOSITY={ env_level_str } , "
f " has to be one of: { ' , ' . join ( log_levels . keys ( ) ) } "
f " has to be one of: { ' , ' . join ( log_levels . keys ( ) ) } "
)
)
return _default_log_level
return _default_log_level
@ -83,7 +83,7 @@ def _configure_library_root_logger() -> None:
_default_handler = logging . StreamHandler ( ) # Set sys.stderr as stream.
_default_handler = logging . StreamHandler ( ) # Set sys.stderr as stream.
_default_handler . flush = sys . stderr . flush
_default_handler . flush = sys . stderr . flush
formatter = logging . Formatter (
formatter = logging . Formatter (
" \033 [1;31m[ %(levelname)s |( BM Tools)%(module)s : %(lineno)d ] %(asctime)s >> \033 [0m %(message)s " )
" \033 [1;31m[ %(levelname)s |( SWARMS Tools)%(module)s : %(lineno)d ] %(asctime)s >> \033 [0m %(message)s " )
_default_handler . setFormatter ( formatter )
_default_handler . setFormatter ( formatter )
# Apply our default configuration to the library root logger.
# Apply our default configuration to the library root logger.
@ -278,7 +278,7 @@ def get_logger(name: Optional[str] = None, verbosity='info') -> logging.Logger:
logger . setLevel ( log_levels [ verbosity ] )
logger . setLevel ( log_levels [ verbosity ] )
# Set up a file handler to write log messages to a file
# Set up a file handler to write log messages to a file
# file_handler = logging.FileHandler('/Users/xuanhe/Documents/our-paper/instructdb/code/BMTools/ bm tools/tools/database/my_log_file.log')
# file_handler = logging.FileHandler('/Users/xuanhe/Documents/our-paper/instructdb/code/BMTools/ swarms. tools/tools/database/my_log_file.log')
# file_handler.setLevel(log_levels[verbosity])
# file_handler.setLevel(log_levels[verbosity])
# logger.addHandler(file_handler)
# logger.addHandler(file_handler)
Zack
1 year ago