Some more caching.

StanfordVL · Sep 2, 2023 · fd3ab50 · fd3ab50
1 parent 67cbe89
commit fd3ab50
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bddl/knowledge_base/models.py b/bddl/knowledge_base/models.py
@@ -1,5 +1,5 @@
 from dataclasses import dataclass, field
-from functools import cached_property
+from functools import cached_property, cache
 import itertools
 import json
 import networkx as nx
@@ -144,6 +144,7 @@ class Meta:
         pk = 'name'
         ordering = ['name']
 
+    @cache
     def matching_synset(self, synset) -> bool:
         return synset.name in self.matching_synsets
 
@@ -178,6 +179,7 @@ class Meta:
         pk = 'name'
         ordering = ['name']
 
+    @cache
     def matching_synset(self, synset) -> bool:
         return self.category.matching_synset(synset)