From 764ee5757e4ef8ff063c8daac054044a10bcc061 Mon Sep 17 00:00:00 2001 From: jalencato Date: Fri, 22 Sep 2023 11:05:20 -0700 Subject: [PATCH] Update heterogeneous_graphloader.py --- .../graph_loaders/heterogeneous_graphloader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/graphstorm-processing/graphstorm_processing/graph_loaders/heterogeneous_graphloader.py b/graphstorm-processing/graphstorm_processing/graph_loaders/heterogeneous_graphloader.py index e1428b7af9..1777ebf085 100644 --- 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