steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-12-10 17:40:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

object and objects are now private

Browse Source
This commit is contained in:
Chanumask
2023-10-23 16:44:03 +02:00
parent e87bd3aaa0
commit c4ffdb4e44
11 changed files with 30 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
marl_factory_grid/environment/groups/objects.py
View File

@@ -3,12 +3,12 @@ from typing import List
import numpy as np
from marl_factory_grid.environment.entity.object import Object
from marl_factory_grid.environment.entity.object import _Object
import marl_factory_grid.environment.constants as c
class Objects:
_entity = Object
class _Objects:
_entity = _Object
@property
def observers(self):
@@ -129,8 +129,8 @@ class Objects:
self.add_items([self._entity() for _ in range(n)])
return c.VALID
def despawn(self, items: List[Object]):
items = [items] if isinstance(items, Object) else items
def despawn(self, items: List[_Object]):
items = [items] if isinstance(items, _Object) else items
for item in items:
del self[item]
@@ -145,7 +145,7 @@ class Objects:
# except (ValueError, AttributeError):
# pass
def notify_del_entity(self, entity: Object):
def notify_del_entity(self, entity: _Object):
try:
entity.del_observer(self)
except AttributeError:
@@ -155,7 +155,7 @@ class Objects:
except (AttributeError, ValueError, IndexError):
pass
def notify_add_entity(self, entity: Object):
def notify_add_entity(self, entity: _Object):
try:
if self not in entity.observers:
entity.add_observer(self)