Added wind scale factor for visualization

2024-01-10 10:31:20 -05:00
parent 045f47b8dd
commit e13f7037fa
3 changed files with 5 additions and 4 deletions
--- a/rotorpy/learning/quadrotor_environments.py
+++ b/rotorpy/learning/quadrotor_environments.py
@@ -165,7 +165,7 @@ class QuadrotorEnv(gym.Env):
                colors = list(mcolors.CSS4_COLORS)
            else:
                colors = [color]
-            self.quad_obj = Quadrotor(self.ax, wind=True, color=np.random.choice(colors))
+            self.quad_obj = Quadrotor(self.ax, wind=True, color=np.random.choice(colors), wind_scale_factor=5)
            self.world_artists = None
            self.title_artist = self.ax.set_title('t = {}'.format(self.t))