steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-06-20 03:08:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge problems

Browse Source
This commit is contained in:
steffen-illium
2021-05-12 12:44:08 +02:00
parent 07ab28ea70
commit 4b24b4f0f2
2 changed files with 26 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
environments/factory/simple_factory_getting_dirty.py
View File

@ -2,15 +2,13 @@ import numpy as np
from environments.factory.base_factory import BaseFactory
from collections import namedtuple
DIRT_INDEX = -1
DirtProperties = namedtuple('DirtProperties', ['clean_amount', 'max_spawn_ratio', 'gain_amount'],
defaults=[0.25, 0.1, 0.1])
class GettingDirty(BaseFactory):
_dirt_indx = -1
def __init__(self, *args, dirt_properties:DirtProperties, **kwargs):
super(GettingDirty, self).__init__(*args, **kwargs)
self._dirt_properties = dirt_properties
@ -21,7 +19,10 @@ class GettingDirty(BaseFactory):
# randomly distribute dirt across the grid
n_dirt_tiles = self._dirt_properties.max_spawn_ratio * len(free_for_dirt)
for x, y in free_for_dirt[:n_dirt_tiles]:
self.state[self._dirt_indx, x, y] += self._dirt_properties.gain_amount
self.state[DIRT_INDEX, x, y] += self._dirt_properties.gain_amount
def additional_actions(self, agent_i, action) -> ((int, int), bool):
if action ==
def reset(self):
# ToDo: When self.reset returns the new states and stuff, use it here!