Use rich click and cleanup (#99)

README.md

@@ -410,7 +410,6 @@ Currently, we have:
 - [depth estimation](mbodied/agents/sense/depth_estimation_agent.py)
 - [object detection](mbodied/agents/sense/object_detection_agent.py)
 - [image segmentation](mbodied/agents/sense/segmentation_agent.py)
-- [3D object pose estimator](mbodied/agents/sense/object_pose_estimator_3d.py)
 
 agents that process robot's sensor information.
 

examples/1_simple_robot_agent.py

@@ -16,7 +16,7 @@
 
 import os
 
-import click
+import rich_click as click
 from pydantic import Field
 
 from mbodied.agents.language import LanguageAgent

examples/3_reason_plan_act_robot.py

@@ -22,7 +22,7 @@
 import os
 from pathlib import Path
 
-import click
+import rich_click as click
 
 from mbodied.agents.language import LanguageAgent
 from mbodied.agents.motion.openvla_agent import OpenVlaAgent

examples/4_language_reason_plan_act_robot.py

@@ -20,7 +20,7 @@
 import os
 from pathlib import Path
 
-import click
+import rich_click as click
 from gymnasium import spaces
 from pydantic import Field
 

examples/5_teach_robot_record_dataset.py

@@ -22,7 +22,7 @@
     python examples/5_teach_robot_record_dataset.py --task "pick up the remote" --backend "openai"
 """
 
-import click
+import rich_click as click
 
 from mbodied.agents.language import LanguageAgent
 from mbodied.data.replaying import Replayer

mbodied/__about__.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "1.2.0"
+__version__ = "1.2.6"

mbodied/data/replaying.py

@@ -17,7 +17,7 @@
 from pathlib import Path
 from typing import Any, Callable, List, Tuple
 
-import click
+import rich_click as click
 import h5py
 import numpy as np
 from datasets import Dataset, DatasetInfo, Features, Image, Value