From 2b4b6e0c238208256f2000ef0f982f91cdea6af8 Mon Sep 17 00:00:00 2001 From: Alejandro Velez-Arce Date: Tue, 24 Sep 2024 09:04:48 -0400 Subject: [PATCH 1/2] Revert "Merge pull request #316 from michellemli/main" This reverts commit 45cc0948d064c6728f28bfc9791468b3e9aabba9, reversing changes made to 80acefc64d5e707d417dcb07b901771e8264dcea. --- tdc/resource/primekg.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tdc/resource/primekg.py b/tdc/resource/primekg.py index 38935b24..ae0df10a 100644 --- 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_id", + "y_id"]].values, + relation=i) return G def get_features(self, feature_type): From 15f46ed9c17039a5d8c52c47faa76f4ca955bcc2 Mon Sep 17 00:00:00 2001 From: Alejandro Velez-Arce Date: Tue, 24 Sep 2024 09:38:08 -0400 Subject: [PATCH 2/2] Revert "Revert "Merge pull request #316 from michellemli/main"" This reverts commit 2b4b6e0c238208256f2000ef0f982f91cdea6af8. --- tdc/resource/primekg.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tdc/resource/primekg.py b/tdc/resource/primekg.py index ae0df10a..38935b24 100644 --- 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_id", - "y_id"]].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):