Improving class loading efficiency and considering dynamic type list

forte/data/data_store.py

@@ -22,7 +22,7 @@
 from sortedcontainers import SortedList
 from typing_inspect import get_origin, get_args, is_generic_type
 
-from forte.utils import get_class
+from forte.utils import get_class, get_class_nc
 from forte.utils.utils import get_full_module_name
 from forte.data.ontology.code_generation_objects import EntryTree
 from forte.data.ontology.ontology_code_generator import OntologyCodeGenerator
@@ -879,7 +879,7 @@ def _is_subclass(
             if cls_qualified_name in type_name_parent_class:
                 return True
             else:
-                entry_class = get_class(type_name)
+                entry_class = get_class_nc(type_name)
                 if issubclass(entry_class, cls):
                     type_name_parent_class.add(cls_qualified_name)
                     return True

forte/utils/utils.py

@@ -23,10 +23,13 @@
 from sortedcontainers import SortedList
 from typing_inspect import is_union_type, get_origin
 
+from functools import lru_cache
+
 __all__ = [
     "get_full_module_name",
     "get_class_name",
     "get_class",
+    "get_class_nc",
     "get_qual_name",
     "create_class_with_kwargs",
     "check_type",
@@ -77,7 +80,14 @@ def get_class_name(o, lower: bool = False) -> str:
         return o.__name__
 
 
+@lru_cache
 def get_class(class_name: str, module_paths: Optional[List[str]] = None):
+    r"""This is the cached version of get_class_nc.
+    """
+    return get_class_nc(str, module_paths)
+
+
+def get_class_nc(class_name: str, module_paths: Optional[List[str]] = None):
     r"""Returns the class based on class name.
 
     Args: