mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-20 22:12:03 +00:00
d337f4e571
* fix double reset, as suggested by @jietang * better floors and ceilings * add convenience methods to monitor * add wrappers to gym namespace * allow playing Atari games, with potentially more coming in the future * simplify example in docs * Move play out of the Env * fix tests * no more deprecation warnings * remove env.monitor * monitor simplification * monitor simplifications * monitor related fixes * a few changes suggested by linter * timestep_limit fixes * keep track of gym env variables for future compatibility * timestep_limit => max_episode_timesteps * don't apply TimeLimit wrapper in make for VNC envs * Respect old timestep_limit argument * Pass max_episode_seconds through registration * Don't include deprecation warnings yet
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
import logging
|
|
import sys
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
root_logger = logging.getLogger()
|
|
|
|
# Should be "gym", but we'll support people doing somewhat crazy
|
|
# things.
|
|
package_name = '.'.join(__name__.split('.')[:-1])
|
|
gym_logger = logging.getLogger(package_name)
|
|
|
|
# Should be modified only by official Gym plugins. This is an
|
|
# unsupported API and may be removed in future versions.
|
|
_extra_loggers = [gym_logger]
|
|
|
|
# Set up the default handler
|
|
formatter = logging.Formatter('[%(asctime)s] %(message)s')
|
|
handler = logging.StreamHandler(sys.stderr)
|
|
handler.setFormatter(formatter)
|
|
|
|
# We need to take in the gym logger explicitly since this is called
|
|
# at initialization time.
|
|
def logger_setup(_=None):
|
|
# This used to take in an argument; we still take an (ignored)
|
|
# argument for compatibility.
|
|
root_logger.addHandler(handler)
|
|
for logger in _extra_loggers:
|
|
logger.setLevel(logging.INFO)
|
|
|
|
def undo_logger_setup():
|
|
"""Undoes the automatic logging setup done by OpenAI Gym. You should call
|
|
this function if you want to manually configure logging
|
|
yourself. Typical usage would involve putting something like the
|
|
following at the top of your script:
|
|
|
|
gym.undo_logger_setup()
|
|
logger = logging.getLogger()
|
|
logger.addHandler(logging.StreamHandler(sys.stderr))
|
|
"""
|
|
root_logger.removeHandler(handler)
|
|
for logger in _extra_loggers:
|
|
logger.setLevel(logging.NOTSET)
|