fix examples

dfki-ric · Feb 10, 2024 · 1ced7fa · 1ced7fa
1 parent 569d65a
commit 1ced7fa
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/examples/floating_mia_example.py b/examples/floating_mia_example.py
@@ -7,8 +7,8 @@
 Floating Mia Example
 =========
 
-This is an example of how to use the FloatingMiaGraspEnv. A random policy is then
-used to generate ten episodes.
+This is an example of how to use the FloatingMiaGraspEnv. A random policy is 
+then used to generate ten episodes.
 
 """
 
@@ -18,7 +18,7 @@
 episode_return = 0
 num_episodes = 0
 
-while num_episodes <= 10:
+while num_episodes < 10:
 
     action = env.action_space.sample()
 
@@ -30,6 +30,6 @@
         num_episodes += 1
         episode_return = 0
 
-        env.reset()
+        obs, _ = env.reset()
 
 env.close()
diff --git a/examples/floating_shadow_example.py b/examples/floating_shadow_example.py
@@ -1,7 +1,6 @@
 import gymnasium
 
-from deformable_gym.envs.floating_shadow_grasp_env import \
-    FloatingShadowGraspEnv
+from deformable_gym.envs.floating_shadow_grasp_env import FloatingShadowGraspEnv
 
 """
 =========