fix formating

mims-harvard · Apr 29, 2024 · c364743 · c364743
1 parent 4b290c3
commit c364743
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tdc/resource/primekg.py b/tdc/resource/primekg.py
@@ -57,7 +57,7 @@ def __init__(self, path="./data"):
         self.df = df
         self.path = path
         super().__init__(self.df)
-    
+
     def get_data(self):
         return self.df
 
@@ -80,4 +80,4 @@ def get_features(self, feature_type):
     def get_node_list(self, node_type):
         df = self.df
         return np.unique(df[(df.x_type == node_type)].x_id.unique().tolist() +
-                         df[(df.y_type == node_type)].y_id.unique().tolist())
+                         df[(df.y_type == node_type)].y_id.unique().tolist())