您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
41 行
1.3 KiB
41 行
1.3 KiB
from mlagents_envs.side_channel.side_channel import SideChannel, SideChannelType
|
|
from typing import List
|
|
|
|
|
|
class RawBytesChannel(SideChannel):
|
|
"""
|
|
This is an example of what the SideChannel for raw bytes exchange would
|
|
look like. Is meant to be used for general research purpose.
|
|
"""
|
|
|
|
def __init__(self, channel_id=0):
|
|
self._received_messages = []
|
|
self._channel_id = channel_id
|
|
super().__init__()
|
|
|
|
@property
|
|
def channel_type(self) -> int:
|
|
return SideChannelType.RawBytesChannelStart + self._channel_id
|
|
|
|
def on_message_received(self, data: bytes) -> None:
|
|
"""
|
|
Is called by the environment to the side channel. Can be called
|
|
multiple times per step if multiple messages are meant for that
|
|
SideChannel.
|
|
"""
|
|
self._received_messages.append(data)
|
|
|
|
def get_and_clear_received_messages(self) -> List[bytearray]:
|
|
"""
|
|
returns a list of bytearray received from the environment.
|
|
"""
|
|
result = list(self._received_messages)
|
|
self._received_messages = []
|
|
return result
|
|
|
|
def send_raw_data(self, data: bytearray) -> None:
|
|
"""
|
|
Queues a message to be sent by the environment at the next call to
|
|
step.
|
|
"""
|
|
super().queue_message_to_send(data)
|