Skip to content

Commit 8e31b36

Browse files
Pabloo22Copilot
andauthored
[Tests] Fix test_all_nodes_are_removed
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
1 parent 00e2ef3 commit 8e31b36

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/reinforcement_learning/test_multi_job_shop_graph_env.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@ def test_all_nodes_are_removed(
155155

156156
removed_nodes = multi_job_shop_graph_env.job_shop_graph.removed_nodes
157157
try:
158-
assert np.all(removed_nodes)
158+
assert all(np.all(lst) for lst in removed_nodes.values())
159159
except AssertionError:
160160
print(removed_nodes)
161161
print(env.instance.to_dict())

0 commit comments

Comments
 (0)