mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-05-22 14:56:43 +02:00
27 lines
643 B
Python
27 lines
643 B
Python
from typing import List
|
|
from marl_factory_grid.environment.rules import Rule
|
|
from marl_factory_grid.utils.results import TickResult, DoneResult
|
|
|
|
|
|
class TemplateRule(Rule):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(TemplateRule, self).__init__()
|
|
self.args = args
|
|
self.kwargs = kwargs
|
|
|
|
def on_init(self, state, lvl_map):
|
|
pass
|
|
|
|
def tick_pre_step(self, state) -> List[TickResult]:
|
|
pass
|
|
|
|
def tick_step(self, state) -> List[TickResult]:
|
|
pass
|
|
|
|
def tick_post_step(self, state) -> List[TickResult]:
|
|
pass
|
|
|
|
def on_check_done(self, state) -> List[DoneResult]:
|
|
pass
|