Update functional.py (dmlc#4015)

anirband · May 17, 2022 · de5543e · de5543e
1 parent bacf2ab
commit de5543e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/dgl/transforms/functional.py b/python/dgl/transforms/functional.py
@@ -21,7 +21,11 @@
 import numpy as np
 import scipy.sparse as sparse
 import scipy.sparse.linalg
-import torch as th
+
+try:
+    import torch as th
+except ImportError:
+    pass
 
 from .._ffi.function import _init_api
 from ..base import dgl_warning, DGLError, NID, EID