pre-commit

Author: nadarenator

pufferlib/ocean/drive/drive.h

@@ -2789,17 +2789,18 @@ void draw_agent_obs(Drive *env, int agent_index, int mode, int obs_only, int las
 
     if (mode == 0) {
         DrawSphere((Vector3){goal_x, goal_y, goal_z}, 0.5f, LIGHTGREEN);
-        DrawCircle3D((Vector3){goal_x, goal_y, goal_z}, env->agents[active_idx].reward_coefs[REWARD_COEF_GOAL_RADIUS], (Vector3){0, 0, 1}, 90.0f,
-                     Fade(LIGHTGREEN, 0.3f));
+        DrawCircle3D((Vector3){goal_x, goal_y, goal_z}, env->agents[active_idx].reward_coefs[REWARD_COEF_GOAL_RADIUS],
+                     (Vector3){0, 0, 1}, 90.0f, Fade(LIGHTGREEN, 0.3f));
     }
 
     if (mode == 1) {
         float goal_x_world = px + (goal_x * heading_self_x - goal_y * heading_self_y);
         float goal_y_world = py + (goal_x * heading_self_y + goal_y * heading_self_x);
         float goal_z_world = pz + goal_z;
         DrawSphere((Vector3){goal_x_world, goal_y_world, goal_z_world}, 0.5f, LIGHTGREEN);
-        DrawCircle3D((Vector3){goal_x_world, goal_y_world, goal_z_world}, env->agents[active_idx].reward_coefs[REWARD_COEF_GOAL_RADIUS],
-                     (Vector3){0, 0, 1}, 90.0f, Fade(LIGHTGREEN, 0.3f));
+        DrawCircle3D((Vector3){goal_x_world, goal_y_world, goal_z_world},
+                     env->agents[active_idx].reward_coefs[REWARD_COEF_GOAL_RADIUS], (Vector3){0, 0, 1}, 90.0f,
+                     Fade(LIGHTGREEN, 0.3f));
     }
     // First draw other agent observations
     int obs_idx = ego_dim; // Start after ego obs