lint

awslabs · Dec 5, 2023 · 3028996 · 3028996
1 parent f8c88e2
commit 3028996
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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)