您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
71 行
3.2 KiB
71 行
3.2 KiB
from sys import platform
|
|
from typing import Optional, Any, List
|
|
from mlagents_envs.environment import UnityEnvironment
|
|
from mlagents_envs.base_env import BaseEnv
|
|
from mlagents_envs.registry.binary_utils import get_local_binary_path
|
|
from mlagents_envs.registry.base_registry_entry import BaseRegistryEntry
|
|
|
|
|
|
class RemoteRegistryEntry(BaseRegistryEntry):
|
|
def __init__(
|
|
self,
|
|
identifier: str,
|
|
expected_reward: Optional[float],
|
|
description: Optional[str],
|
|
linux_url: Optional[str],
|
|
darwin_url: Optional[str],
|
|
win_url: Optional[str],
|
|
additional_args: Optional[List[str]] = None,
|
|
):
|
|
"""
|
|
A RemoteRegistryEntry is an implementation of BaseRegistryEntry that uses a
|
|
Unity executable downloaded from the internet to launch a UnityEnvironment.
|
|
__Note__: The url provided must be a link to a `.zip` file containing a single
|
|
compressed folder with the executable inside. There can only be one executable
|
|
in the folder and it must be at the root of the folder.
|
|
:param identifier: The name of the Unity Environment.
|
|
:param expected_reward: The cumulative reward that an Agent must receive
|
|
for the task to be considered solved.
|
|
:param description: A description of the Unity Environment. Contains human
|
|
readable information about potential special arguments that the make method can
|
|
take as well as information regarding the observation, reward, actions,
|
|
behaviors and number of agents in the Environment.
|
|
:param linux_url: The url of the Unity executable for the Linux platform
|
|
:param darwin_url: The url of the Unity executable for the OSX platform
|
|
:param win_url: The url of the Unity executable for the Windows platform
|
|
"""
|
|
super().__init__(identifier, expected_reward, description)
|
|
self._linux_url = linux_url
|
|
self._darwin_url = darwin_url
|
|
self._win_url = win_url
|
|
self._add_args = additional_args
|
|
|
|
def make(self, **kwargs: Any) -> BaseEnv:
|
|
"""
|
|
Returns the UnityEnvironment that corresponds to the Unity executable found at
|
|
the provided url. The arguments passed to this method will be passed to the
|
|
constructor of the UnityEnvironment (except for the file_name argument)
|
|
"""
|
|
url = None
|
|
if platform == "linux" or platform == "linux2":
|
|
url = self._linux_url
|
|
if platform == "darwin":
|
|
url = self._darwin_url
|
|
if platform == "win32":
|
|
url = self._win_url
|
|
if url is None:
|
|
raise FileNotFoundError(
|
|
f"The entry {self.identifier} does not contain a valid url for this "
|
|
"platform"
|
|
)
|
|
path = get_local_binary_path(self.identifier, url)
|
|
if "file_name" in kwargs:
|
|
kwargs.pop("file_name")
|
|
args: List[str] = []
|
|
if "additional_args" in kwargs:
|
|
if kwargs["additional_args"] is not None:
|
|
args += kwargs["additional_args"]
|
|
if self._add_args is not None:
|
|
args += self._add_args
|
|
kwargs["additional_args"] = args
|
|
return UnityEnvironment(file_name=path, **kwargs)
|