Skip to content

Commit a61bfce

Browse files
committed
fix: nodes as the well-defined variable
1 parent b974ffd commit a61bfce

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

spras/dataset.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -104,11 +104,11 @@ def load_files_from_dict(self, dataset_dict):
104104
f"Edge file {interactome_loc} must have three or four columns but found {num_cols}"
105105
)
106106

107-
node_set = set(self.interactome.Interactor1.unique())
108-
node_set = node_set.union(set(self.interactome.Interactor2.unique()))
107+
# We get uniqueness afterwards to make `load_files_from_dict` have a well-defined node ordering
108+
nodes = list(pd.concat([self.interactome.Interactor1, self.interactome.Interactor2]).unique())
109109

110110
# Load generic node tables
111-
self.node_table = pd.DataFrame(node_set, columns=[self.NODE_ID])
111+
self.node_table = pd.DataFrame(nodes, columns=[self.NODE_ID])
112112
for node_file in node_data_files:
113113
single_node_table = pd.read_table(os.path.join(data_loc, node_file))
114114
# If we have only 1 column, assume this is an indicator variable

0 commit comments

Comments
 (0)