Use list comprehensions

Author: natoverse

graphrag/index/flows/create_base_entity_graph.py

@@ -168,7 +168,6 @@ def _prep_communities(communities) -> pd.DataFrame:
 
 
 def _compute_degree(graph: nx.Graph) -> pd.DataFrame:
-    degrees = []
-    for node, degree in graph.degree:  # type: ignore
-        degrees.append({"name": node, "degree": int(degree)})
-    return pd.DataFrame(degrees)
+    return pd.DataFrame([
+        {"name": node, "degree": int(degree)} for node, degree in graph.degree
+    ])  # type: ignore

graphrag/index/flows/create_final_communities.py

@@ -60,7 +60,7 @@ def create_final_communities(
 
     # join it all up and add some new fields
     communities = all_grouped.merge(entity_ids, on="community", how="inner")
-    communities["id"] = communities["community"].apply(lambda _x: str(uuid4()))
+    communities["id"] = [str(uuid4()) for _ in range(len(communities))]
     communities["human_readable_id"] = communities["community"]
     communities["title"] = "Community " + communities["community"].astype(str)