update: test_initialize_q_expressions

kozistr · Feb 1, 2025 · 6772208 · 6772208
1 parent b18af3f
commit 6772208
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -325,7 +325,7 @@ def test_update_precondition_dense():
 
 def test_initialize_q_expressions():
     x = torch.zeros(1)
-    _ = initialize_q_expressions(x, 0.0, 0, 0, None)
+    _ = initialize_q_expressions(x.squeeze(), 0.0, 0, 0, None)
 
     with pytest.raises(ValueError):
         initialize_q_expressions(x.expand(1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1), 0.0, 0, 0, None)