fix existing test

bdaiinstitute · Jul 18, 2023 · 544bd75 · 544bd75
1 parent 4d786fe
commit 544bd75
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/envs/test_kitchen.py b/tests/envs/test_kitchen.py
@@ -35,12 +35,13 @@ def test_kitchen():
         task = perceiver.reset(env_task)
         for obj in task.init:
             assert len(obj.type.feature_names) == len(task.init[obj])
-    assert len(env.predicates) == 3
-    At, OnTop, TurnedOn = sorted(env.predicates)
+    assert len(env.predicates) == 4
+    At, CanTurnDial, OnTop, TurnedOn = sorted(env.predicates)
     assert At.name == "At"
+    assert CanTurnDial.name == "CanTurnDial"
     assert OnTop.name == "OnTop"
     assert TurnedOn.name == "TurnedOn"
-    assert env.goal_predicates == {At, OnTop, TurnedOn}
+    assert env.goal_predicates == {At, CanTurnDial, OnTop, TurnedOn}
     options = get_gt_options(env.get_name())
     assert len(options) == 3
     moveto_option, pushobjonobjforward_option, pushobjturnonright_option = \