[BugFix] Rotation dynamics

Author: matteobettini

vmas/simulator/dynamics/roatation.py

@@ -11,4 +11,4 @@ def needed_action_size(self) -> int:
         return 1
 
     def process_action(self):
-        self.agent.state.torque = self.agent.action.u[:, 0]
+        self.agent.state.torque = self.agent.action.u[:, 0].unsqueeze(-1)