Add function to get required meta links for synset

StanfordVL · Dec 3, 2024 · b4816b2 · b4816b2
1 parent 5288b8c
commit b4816b2
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/bddl/object_taxonomy.py b/bddl/object_taxonomy.py
@@ -320,6 +320,16 @@ def get_required_meta_links_for_abilities(abilities):
 
         return required_links
 
+    def get_required_meta_links_for_synset(self, synset):
+        """
+        Get the required meta links for a given synset.
+
+        :param synset: synset to search.
+        :return: set of str containing required meta links.
+        """
+        abilities = self.get_abilities(synset)
+        return self.get_required_meta_links_for_abilities(abilities)
+
 
 if __name__ == "__main__":
     object_taxonomy = ObjectTaxonomy()