useOfAi

RL Testing/lev_pod_env.py

@@ -155,6 +155,7 @@ class LevPodEnv(gym.Env):
         
         return obs, {}
 
+    # The following was generated by AI - see [14]
     def step(self, action):
         # Check if PyBullet connection is still active (GUI might be closed)
         try:
@@ -374,6 +375,7 @@ class LevPodEnv(gym.Env):
         
         return obs, reward, terminated, truncated, info
 
+    # The following was generated by AI - see [15]
     def _get_obs(self, initial_reset=False):
         """
         Returns observation: [gaps(4), velocities(4)]
@@ -433,6 +435,7 @@ class LevPodEnv(gym.Env):
         
         return obs
     
+    # The following was generated by AI - see [16]
     def _create_modified_urdf(self):
         """
         Create a modified URDF with bolt positions adjusted based on initial gap height.