From 9d9c7063cd49bb19ad216efdb9b20be4d52903d9 Mon Sep 17 00:00:00 2001
From: Leni Aniva <v@leni.sh>
Date: Fri, 4 Oct 2024 18:42:33 -0700
Subject: [PATCH] feat: Remove the goal count restriction on initial state

---
 pantograph/search.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/pantograph/search.py b/pantograph/search.py
index cb87a70..7aa3342 100644
--- a/pantograph/search.py
+++ b/pantograph/search.py
@@ -80,13 +80,12 @@ class Agent:
         Searches using th
         """
         assert server.is_automatic(), "Search must be run in automatic mode"
-        assert len(goal_state.goals) == 1, "Initial state must have exactly one goal"
 
         initial_state = SearchState(
             state=goal_state,
             parent=None,
             parent_goal_id=None,
-            priorities=[0.0]
+            priorities=[0.0 for _ in goal_state.goals]
         )
         search_stack = [initial_state]
         """