Minor fixes

mujoco_maze/ant.py

@@ -16,7 +16,7 @@
 """Wrapper for creating the ant environment in gym_mujoco."""
 
 import math
-from typing import Tuple
+from typing import Optional, Tuple
 
 import numpy as np
 

mujoco_maze/maze_env.py

@@ -95,7 +95,7 @@ class MazeEnv(gym.Env):
             # Increase initial z-pos of ant.
             height_offset = height * size_scaling
             torso = tree.find(".//body[@name='torso']")
-            torso.set("pos", "0 0 %.2f" % (0.75 + height_offset))
+            torso.set("pos", f"0 0 {0.75 + height_offset:%.2f}")
         if self.blocks:
             # If there are movable blocks, change simulation settings to perform
             # better contact detection.

mujoco_maze/point.py

@@ -25,7 +25,7 @@ from mujoco_maze.agent_model import AgentModel
 
 
 class PointEnv(AgentModel):
-    VELOCITY_LIMITS: float = 100.0
+    VELOCITY_LIMITS: float = 10.0
     FILE = "point.xml"
     ORI_IND = 2