From 4ea3cbf364169b73ee5278647ceb85d5ff02fd73 Mon Sep 17 00:00:00 2001
From: Ephraim Rusu <rusu1@llnl.gov>
Date: Tue, 24 Oct 2023 09:22:06 -0700
Subject: [PATCH] udpated registry

---
 abmarl/sim/gridworld/registry.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/abmarl/sim/gridworld/registry.py b/abmarl/sim/gridworld/registry.py
index 1de3dd7b..9da02f34 100644
--- 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
     }
 }