From a8b69d1a0ef1b417758cd963ced6839aeddab8d3 Mon Sep 17 00:00:00 2001
From: Eric Seuret <eric@ericst.ch>
Date: Tue, 30 Sep 2025 11:56:58 +0200
Subject: [PATCH] Max speed

---
 WheeledRobot.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/WheeledRobot.py b/WheeledRobot.py
index 48b81d9..f9d5797 100755
--- a/WheeledRobot.py
+++ b/WheeledRobot.py
@@ -141,6 +141,9 @@ class WheeledRobotEnv(gym.Env):
         # U represents wheel velocities from controllers
         # F converts wheel velocities to robot velocities
         robot_velocities = self.F @ U
+        robot_velocities = np.clip(robot_velocities, 
+                                   a_min=np.array([-2, -2, -1]).reshape(-1, 1), 
+                                   a_max=np.array([2, 2, 1]).reshape(-1, 1))
         Xp = X + self._R(X[2, 0]) @ robot_velocities * self.dt
         
         previous_state = self.state.copy()