Merge pull request #450 from LLNL/abmarl-449-update-registry

udpated registry
LLNL · Oct 24, 2023 · 7de2630 · 7de2630
2 parents c321f0c + 4ea3cbf
commit 7de2630
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/abmarl/sim/gridworld/registry.py b/abmarl/sim/gridworld/registry.py
@@ -1,12 +1,13 @@
 
 from .actor import ActorBaseComponent, MoveActor, CrossMoveActor, BinaryAttackActor, \
     EncodingBasedAttackActor, RestrictedSelectiveAttackActor, SelectiveAttackActor
-from .done import DoneBaseComponent, ActiveDone, TargetAgentDone, OneTeamRemainingDone
+from .done import DoneBaseComponent, ActiveDone, TargetAgentDone, OneTeamRemainingDone, \
+    TargetDestroyedDone
 from .observer import ObserverBaseComponent, AbsoluteEncodingObserver, \
     PositionCenteredEncodingObserver, StackedPositionCenteredEncodingObserver, \
-    AbsolutePositionObserver
+    AbsolutePositionObserver, AmmoObserver
 from .state import StateBaseComponent, PositionState, TargetBarriersFreePlacementState, \
-    MazePlacementState, HealthState
+    MazePlacementState, HealthState, AmmoState
 
 
 _subclass_check_mapping = {
@@ -27,17 +28,20 @@
     }, 'done': {
         ActiveDone,
         TargetAgentDone,
+        TargetDestroyedDone,
         OneTeamRemainingDone
     }, 'observer': {
         AbsoluteEncodingObserver,
         PositionCenteredEncodingObserver,
         StackedPositionCenteredEncodingObserver,
-        AbsolutePositionObserver
+        AbsolutePositionObserver,
+        AmmoObserver
     }, 'state': {
         PositionState,
         TargetBarriersFreePlacementState,
         MazePlacementState,
-        HealthState
+        HealthState,
+        AmmoState
     }
 }