mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-09-18 00:21:58 +02:00
WIP: object, entity rework
This commit is contained in:
@@ -1,9 +1,9 @@
|
||||
from marl_factory_grid.environment.groups.env_objects import EnvObjects
|
||||
from marl_factory_grid.environment.groups.env_objects import Collection
|
||||
from marl_factory_grid.environment.groups.mixins import PositionMixin, HasBoundMixin
|
||||
from marl_factory_grid.modules.batteries.entitites import Pod, Battery
|
||||
|
||||
|
||||
class Batteries(HasBoundMixin, EnvObjects):
|
||||
class Batteries(Collection):
|
||||
|
||||
_entity = Battery
|
||||
is_blocking_light: bool = False
|
||||
@@ -21,7 +21,7 @@ class Batteries(HasBoundMixin, EnvObjects):
|
||||
self.add_items(batteries)
|
||||
|
||||
|
||||
class ChargePods(PositionMixin, EnvObjects):
|
||||
class ChargePods(PositionMixin, Collection):
|
||||
|
||||
_entity = Pod
|
||||
|
||||
|
Reference in New Issue
Block a user