mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-07-05 17:11:35 +02:00
Merge branch 'main' into unit_testing
This commit is contained in:
@ -142,6 +142,7 @@ class Factory(gym.Env):
|
||||
|
||||
# All is set up, trigger entity spawn with variable pos
|
||||
self.state.rules.do_all_reset(self.state)
|
||||
self.state.rules.do_all_post_spawn_reset(self.state)
|
||||
|
||||
# Build initial observations for all agents
|
||||
self.obs_builder.reset(self.state)
|
||||
@ -218,8 +219,7 @@ class Factory(gym.Env):
|
||||
# Combine Info dicts into a global one
|
||||
combined_info_dict = defaultdict(lambda: 0.0)
|
||||
for result in chain(tick_results, done_check_results):
|
||||
if not result:
|
||||
raise ValueError()
|
||||
assert result, 'Something returned None...'
|
||||
if result.reward is not None:
|
||||
try:
|
||||
rewards[result.entity.name] += result.reward
|
||||
|
Reference in New Issue
Block a user