您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
78 行
2.3 KiB
78 行
2.3 KiB
import logging
|
|
|
|
logger = logging.getLogger("mlagents.envs")
|
|
|
|
|
|
class UnityException(Exception):
|
|
"""
|
|
Any error related to ml-agents environment.
|
|
"""
|
|
|
|
pass
|
|
|
|
|
|
class UnityEnvironmentException(UnityException):
|
|
"""
|
|
Related to errors starting and closing environment.
|
|
"""
|
|
|
|
pass
|
|
|
|
|
|
class UnityActionException(UnityException):
|
|
"""
|
|
Related to errors with sending actions.
|
|
"""
|
|
|
|
pass
|
|
|
|
|
|
class UnityTimeOutException(UnityException):
|
|
"""
|
|
Related to errors with communication timeouts.
|
|
"""
|
|
|
|
def __init__(self, message, log_file_path=None):
|
|
if log_file_path is not None:
|
|
try:
|
|
with open(log_file_path, "r") as f:
|
|
printing = False
|
|
unity_error = "\n"
|
|
for line in f:
|
|
line = line.strip()
|
|
if (line == "Exception") or (line == "Error"):
|
|
printing = True
|
|
unity_error += "----------------------\n"
|
|
if line == "":
|
|
printing = False
|
|
if printing:
|
|
unity_error += line + "\n"
|
|
logger.info(unity_error)
|
|
logger.error(
|
|
"An error might have occured in the environment. "
|
|
"You can check the logfile for more information at {}".format(
|
|
log_file_path
|
|
)
|
|
)
|
|
except Exception:
|
|
logger.error(
|
|
"An error might have occured in the environment. "
|
|
"No UnitySDK.log file could be found."
|
|
)
|
|
super(UnityTimeOutException, self).__init__(message)
|
|
|
|
|
|
class UnityWorkerInUseException(UnityException):
|
|
"""
|
|
This error occurs when the port for a certain worker ID is already reserved.
|
|
"""
|
|
|
|
MESSAGE_TEMPLATE = (
|
|
"Couldn't start socket communication because worker number {} is still in use. "
|
|
"You may need to manually close a previously opened environment "
|
|
"or use a different worker number."
|
|
)
|
|
|
|
def __init__(self, worker_id):
|
|
message = self.MESSAGE_TEMPLATE.format(str(worker_id))
|
|
super(UnityWorkerInUseException, self).__init__(message)
|