From 030bbe72ff4d0a39c68460ed16da623c559e352d Mon Sep 17 00:00:00 2001
From: Naoki Yokoyama <nyokoyama@theaiinstitute.com>
Date: Fri, 1 Sep 2023 13:15:29 -0400
Subject: [PATCH] updating aliases for MP3D object names

---
 zsos/policy/habitat_policies.py |  6 +++---
 zsos/vlm/classes.txt            | 11 ++---------
 2 files changed, 5 insertions(+), 12 deletions(-)

diff --git a/zsos/policy/habitat_policies.py b/zsos/policy/habitat_policies.py
index 95c39fc..7e57cc5 100644
--- a/zsos/policy/habitat_policies.py
+++ b/zsos/policy/habitat_policies.py
@@ -26,13 +26,13 @@
 HM3D_ID_TO_NAME = ["chair", "bed", "potted plant", "toilet", "tv", "couch"]
 MP3D_ID_TO_NAME = [
     "chair",
-    "table",
+    "dining table",  # "table",
     "picture",
     "cabinet",
-    "cushion",
+    "pillow",  # "cushion",
     "couch",  # "sofa",
     "bed",
-    "chest of drawers",
+    "dresser",  # "chest of drawers",
     "potted plant",  # "plant",
     "sink",
     "toilet",
diff --git a/zsos/vlm/classes.txt b/zsos/vlm/classes.txt
index e3341b1..46666aa 100644
--- a/zsos/vlm/classes.txt
+++ b/zsos/vlm/classes.txt
@@ -1,17 +1,10 @@
-chair
-table
 picture
 cabinet
-cushion
-couch
-bed
-chest of drawers
-potted plant
+pillow
+dresser
 sink
-toilet
 stool
 towel
-tv
 shower
 bathtub
 counter