Update heterogeneous_graphloader.py

awslabs · Sep 22, 2023 · 764ee57 · 764ee57
1 parent b1c25dc
commit 764ee57
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/graphstorm-processing/graphstorm_processing/graph_loaders/heterogeneous_graphloader.py b/graphstorm-processing/graphstorm_processing/graph_loaders/heterogeneous_graphloader.py
@@ -14,7 +14,7 @@
 limitations under the License.
 """
 from .loader_base import GraphLoader
-
+import numpy
 
 class HeterogeneousGraphLoader(GraphLoader):
     """DGL Heterogeneous Graph Loader