From 3028996995c3cf50ec45c0fcae45daf3c764ae20 Mon Sep 17 00:00:00 2001 From: Zichen Wang Date: Tue, 5 Dec 2023 22:17:41 +0000 Subject: [PATCH] lint --- python/graphstorm/model/utils.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/graphstorm/model/utils.py b/python/graphstorm/model/utils.py index 5a44bb5d62..6396d42669 100644 --- a/python/graphstorm/model/utils.py +++ b/python/graphstorm/model/utils.py @@ -918,8 +918,10 @@ def load_gsgnn_embeddings(emb_path, g): ntype_emb_path = os.path.join(emb_path, ntype) emb_files = os.listdir(ntype_emb_path) - ntype_emb_files = [file for file in emb_files if file.endswith(".pt") and file.startswith("emb")] - ntype_nid_files = [file for file in emb_files if file.endswith(".pt") and file.startswith("nids")] + ntype_emb_files = [file for file in emb_files if file.endswith(".pt") and + file.startswith("emb")] + ntype_nid_files = [file for file in emb_files if file.endswith(".pt") and + file.startswith("nids")] ntype_emb_files = sorted(ntype_emb_files) ntype_nid_files = sorted(ntype_nid_files) part_policy = g.get_node_partition_policy(ntype)