您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
37 行
782 B
37 行
782 B
from abc import ABC, abstractmethod
|
|
from typing import Dict
|
|
|
|
from mlagents.envs import AllBrainInfo, BrainParameters
|
|
|
|
|
|
class BaseUnityEnvironment(ABC):
|
|
@abstractmethod
|
|
def step(
|
|
self, vector_action=None, memory=None, text_action=None, value=None
|
|
) -> AllBrainInfo:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def reset(
|
|
self, config=None, train_mode=True, custom_reset_parameters=None
|
|
) -> AllBrainInfo:
|
|
pass
|
|
|
|
@property
|
|
@abstractmethod
|
|
def global_done(self):
|
|
pass
|
|
|
|
@property
|
|
@abstractmethod
|
|
def external_brains(self) -> Dict[str, BrainParameters]:
|
|
pass
|
|
|
|
@property
|
|
@abstractmethod
|
|
def reset_parameters(self) -> Dict[str, float]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def close(self):
|
|
pass
|