gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-09-01 02:17:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for configuring monitor output name (#379)

Browse Source 
* configurable monitor filename

* removing debug logging

* Add documentation for uid argument to monitor.start
This commit is contained in:
jietang
2016-10-18 11:37:29 -07:00
committed by GitHub
parent d35c2f0480
commit cc0690cf1a
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
gym/monitoring/monitor.py
View File

@@ -100,7 +100,8 @@ class Monitor(object):
raise error.Error("env has been garbage collected. To keep using a monitor, you must keep around a reference to the env object. (HINT: try assigning the env to a variable in your code.)") raise error.Error("env has been garbage collected. To keep using a monitor, you must keep around a reference to the env object. (HINT: try assigning the env to a variable in your code.)")
return env return env
def start(self, directory, video_callable=None, force=False, resume=False, seed=None, write_upon_reset=False): def start(self, directory, video_callable=None, force=False, resume=False,
seed=None, write_upon_reset=False, uid=None):
"""Start monitoring. """Start monitoring.
Args: Args:
@@ -110,6 +111,7 @@ class Monitor(object):
resume (bool): Retain the training data already in this directory, which will be merged with our new data resume (bool): Retain the training data already in this directory, which will be merged with our new data
seed (Optional[int]): The seed to run this environment with. By default, a random seed will be chosen. seed (Optional[int]): The seed to run this environment with. By default, a random seed will be chosen.
write_upon_reset (bool): Write the manifest file on each reset. (This is currently a JSON file, so writing it is somewhat expensive.) write_upon_reset (bool): Write the manifest file on each reset. (This is currently a JSON file, so writing it is somewhat expensive.)
uid (Optional[str]): A unique id used as part of the suffix for the file. By default, uses os.getpid().
""" """
if self.env.spec is None: if self.env.spec is None:
logger.warn("Trying to monitor an environment which has no 'spec' set. This usually means you did not create it via 'gym.make', and is recommended only for advanced users.") logger.warn("Trying to monitor an environment which has no 'spec' set. This usually means you did not create it via 'gym.make', and is recommended only for advanced users.")
@@ -143,7 +145,8 @@ class Monitor(object):
# We use the 'openai-gym' prefix to determine if a file is # We use the 'openai-gym' prefix to determine if a file is
# ours # ours
self.file_prefix = FILE_PREFIX self.file_prefix = FILE_PREFIX
self.file_infix = '{}.{}'.format(self._monitor_id, os.getpid()) self.file_infix = '{}.{}'.format(self._monitor_id, uid if uid else os.getpid())
self.stats_recorder = stats_recorder.StatsRecorder(directory, '{}.episode_batch.{}'.format(self.file_prefix, self.file_infix)) self.stats_recorder = stats_recorder.StatsRecorder(directory, '{}.episode_batch.{}'.format(self.file_prefix, self.file_infix))
self.configure(video_callable=video_callable) self.configure(video_callable=video_callable)
if not os.path.exists(directory): if not os.path.exists(directory):