Update ml-agents-envs/mlagents_envs/environment.py

Co-authored-by: Vincent-Pierre BERGES <vincentpierre@unity3d.com>

ml-agents-envs/mlagents_envs/environment.py

                 )
             ) from ie
         if action_spec.continuous_size > 0:
-            self._env_actions[behavior_name].continuous[index] = action.continuous[0]
+            self._env_actions[behavior_name].continuous[index] = action.continuous[0, :]
-            self._env_actions[behavior_name].discrete[index] = action.discrete[0]
+            self._env_actions[behavior_name].discrete[index] = action.discrete[0, :]
 
     def get_steps(
         self, behavior_name: BehaviorName