fix(nyz): fix ensemble head unittest bug

Author: PaParaZz1

ding/model/common/tests/test_head.py

@@ -90,4 +90,4 @@ def test_ensemble(self):
         model = EnsembleHead(embedding_dim, action_shape, 3, 3, 3)
         outputs = model(inputs)['pred']
         self.output_check(model, outputs)
-        assert outputs.shape == (B, action_shape * 3, 1)
+        assert outputs.shape == (B, action_shape * 3)