Fix formatting for yapf

mims-harvard · Sep 24, 2024 · bf72829 · bf72829
1 parent 0b6f6e3
commit bf72829
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tdc/resource/primekg.py b/tdc/resource/primekg.py
@@ -33,9 +33,9 @@ def to_nx(self):
 
         G = nx.Graph()
         for i in self.df.relation.unique():
-            G.add_edges_from(self.df[self.df.relation == i][["x_name",
-                                                             "y_name"]].values,
-                             relation=i)
+            G.add_edges_from(
+                self.df[self.df.relation == i][["x_name", "y_name"]].values,
+                relation=i)
         return G
 
     def get_features(self, feature_type):