ray/rllib/examples/env/utils/interfaces.py

23 lines
554 B
Python

##########
# Contribution by the Center on Long-Term Risk:
# https://github.com/longtermrisk/marltoolbox
##########
from abc import ABC, abstractmethod
class InfoAccumulationInterface(ABC):
@abstractmethod
def _init_info(self):
raise NotImplementedError()
@abstractmethod
def _reset_info(self):
raise NotImplementedError()
@abstractmethod
def _get_episode_info(self):
raise NotImplementedError()
@abstractmethod
def _accumulate_info(self, *args, **kwargs):
raise NotImplementedError()