self.actions should be ndarray

issp-center-dev · Dec 23, 2024 · 9109384 · 9109384
1 parent 37205c7
commit 9109384
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/physbo/search/discrete_multi/policy.py b/physbo/search/discrete_multi/policy.py
@@ -58,7 +58,7 @@ def __init__(
                 msg = "ERROR: len(initial_data[0]) != initial_data[1].shape[0]"
                 raise RuntimeError(msg)
             self.write(actions, fs)
-            self.actions = sorted(list(set(self.actions) - set(actions)))
+            self.actions = np.array(sorted(list(set(self.actions) - set(actions))))
 
         if comm is None:
             self.mpicomm = None