Make heuristic agents deterministic

mal-lang · Jan 29, 2025 · b3e896d · b3e896d
1 parent d620792
commit b3e896d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/malsim/agents/heuristic_agent.py b/malsim/agents/heuristic_agent.py
@@ -38,6 +38,9 @@ def get_next_action(
         selected_node_cost = math.inf
         selected_node = None
 
+        # To make it deterministic
+        agent.action_surface.sort(key=lambda n: n.id)
+
         for node in agent.action_surface:
             node_cost = node.extras.get('reward', 0)
 
@@ -85,6 +88,9 @@ def get_next_action(
         selected_node_cost = math.inf
         selected_node = None
 
+        # To make it deterministic
+        agent.action_surface.sort(key=lambda n: n.id)
+
         for node in agent.action_surface:
             node_cost = node.extras.get('reward', 0)