Fix bug in hierarchy generation

StanfordVL · Jan 18, 2024 · 6848e67 · 6848e67
1 parent b7e78fa
commit 6848e67
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/bddl/data_generation/get_hierarchy_full.py b/bddl/data_generation/get_hierarchy_full.py
@@ -23,6 +23,7 @@ def add_igibson_objects(node, synset_to_cat, synset_to_substance):
   '''
   if node["name"] in synset_to_cat:
     node["categories"] = sorted(synset_to_cat[node["name"]])
+  if node["name"] in synset_to_substance:
     node["substances"] = sorted(synset_to_substance[node["name"]])
 
   if "children" in node: