From 6772208420a7b2b0f55171bc0f9b550b5933ff35 Mon Sep 17 00:00:00 2001 From: kozistr Date: Sat, 1 Feb 2025 21:36:01 +0900 Subject: [PATCH] update: test_initialize_q_expressions --- tests/test_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_utils.py b/tests/test_utils.py index baff3e40..db72d7fd 100644 --- 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)