mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-01 14:10:30 +00:00
cdd212db4b
* Fix autodetect dtype warnings * Use warnings module for gym logger * Fix warning in tests
38 lines
711 B
Python
38 lines
711 B
Python
import warnings
|
|
|
|
from gym.utils import colorize
|
|
|
|
DEBUG = 10
|
|
INFO = 20
|
|
WARN = 30
|
|
ERROR = 40
|
|
DISABLED = 50
|
|
|
|
MIN_LEVEL = 30
|
|
|
|
def set_level(level):
|
|
"""
|
|
Set logging threshold on current logger.
|
|
"""
|
|
global MIN_LEVEL
|
|
MIN_LEVEL = level
|
|
|
|
def debug(msg, *args):
|
|
if MIN_LEVEL <= DEBUG:
|
|
print('%s: %s'%('DEBUG', msg % args))
|
|
|
|
def info(msg, *args):
|
|
if MIN_LEVEL <= INFO:
|
|
print('%s: %s'%('INFO', msg % args))
|
|
|
|
def warn(msg, *args):
|
|
if MIN_LEVEL <= WARN:
|
|
warnings.warn(colorize('%s: %s'%('WARN', msg % args), 'yellow'))
|
|
|
|
def error(msg, *args):
|
|
if MIN_LEVEL <= ERROR:
|
|
print(colorize('%s: %s'%('ERROR', msg % args), 'red'))
|
|
|
|
# DEPRECATED:
|
|
setLevel = set_level
|