Merge pull request #1452 from kuzudb/pyg-hot-fix

On demand import for PyG
kuzudb · Apr 6, 2023 · a6f98ac · a6f98ac
2 parents ad464cc + 3441f7b
commit a6f98ac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/python_api/src_py/database.py b/tools/python_api/src_py/database.py
@@ -1,6 +1,4 @@
 from . import _kuzu
-from .torch_geometric_feature_store import KuzuFeatureStore
-from .torch_geometric_graph_store import KuzuGraphStore
 from .types import Type
 
 
@@ -83,6 +81,8 @@ def set_logging_level(self, level):
         self._database.set_logging_level(level)
 
     def get_torch_geometric_remote_backend(self, num_threads=None):
+        from .torch_geometric_feature_store import KuzuFeatureStore
+        from .torch_geometric_graph_store import KuzuGraphStore
         """
         Get the torch_geometric remote backend.