From a8e8766be9ae515a164bd28c45b1a7f1e9514f50 Mon Sep 17 00:00:00 2001 From: Matteo Bettini Date: Tue, 22 Apr 2025 15:39:20 +0200 Subject: [PATCH] amend --- vmas/scenarios/football.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/vmas/scenarios/football.py b/vmas/scenarios/football.py index 023b1ede..43303eb9 100644 --- a/vmas/scenarios/football.py +++ b/vmas/scenarios/football.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2024. +# Copyright (c) 2022-2025. # ProrokLab (https://www.proroklab.org/) # All rights reserved. @@ -1091,7 +1091,9 @@ def process_action(self, agent: Agent): shoot_force = torch.zeros( self.world.batch_dim, 2, device=self.world.device, dtype=torch.float32 ) - shoot_force[..., X] = agent.action.u[..., -1] + self.u_shoot_multiplier + shoot_force[..., X] = ( + agent.action.u[..., -1] * 2.67 * self.u_shoot_multiplier + ) shoot_force = TorchUtils.rotate_vector(shoot_force, agent.state.rot) agent.shoot_force = shoot_force shoot_force = torch.where(