Update test/test_cost.py

Co-authored-by: Vincent Moens <[email protected]>
pytorch · Nov 6, 2023 · c5341c3 · c5341c3
1 parent 20bbd8a
commit c5341c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_cost.py b/test/test_cost.py
@@ -5184,7 +5184,7 @@ def _create_mock_actor(
         elif action_spec_type == "categorical":
             action_spec = DiscreteTensorSpec(action_dim)
         else:
-            raise ValueError(f"Wrong {action_spec_type}")
+            raise ValueError(f"Wrong action spec type: {action_spec_type}")
 
         module = nn.Linear(obs_dim, action_dim)
         if is_nn_module: