random sample needs sequence (in rocksample)

h2r · Jan 27, 2024 · 3c6b8a0 · 3c6b8a0
1 parent 464ebbc
commit 3c6b8a0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pomdp_py/problems/rocksample/rocksample_problem.py b/pomdp_py/problems/rocksample/rocksample_problem.py
@@ -345,7 +345,7 @@ def argmax(self, state, normalized=False, **kwargs):
     def get_all_actions(self, **kwargs):
         state = kwargs.get("state", None)
         if state is None:
-            return self._all_actions
+            return list(self._all_actions)
         else:
             motions = set(self._all_actions)
             rover_x, rover_y = state.position
@@ -355,7 +355,7 @@ def get_all_actions(self, **kwargs):
                 motions.remove(MoveNorth)
             if rover_y == self._n - 1:
                 motions.remove(MoveSouth)
-            return motions | self._other_actions
+            return list(motions | self._other_actions)
 
     def rollout(self, state, history=None):
         return random.sample(self.get_all_actions(state=state), 1)[0]
@@ -529,7 +529,7 @@ def main():
 
     print("*** Testing POMCP ***")
     pomcp = pomdp_py.POMCP(
-        max_depth=20,
+        max_depth=12,
         discount_factor=0.95,
         num_sims=10000,
         exploration_const=20,