Skip to content

Commit ff31bbb

Browse files
committed
fix: row_normalized
1 parent 877ded6 commit ff31bbb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

graph_datasets/load_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,6 @@ def load_data(
151151
f"https://galogm.github.io/graph_datasets_docs/rst/table.html"
152152
)
153153

154-
graph.row_normalized = False
155154
if row_normalize:
156155
graph.ndata["feat"] = F.normalize(graph.ndata["feat"], dim=1)
157156
graph.row_normalized = True
@@ -170,6 +169,7 @@ def load_data(
170169

171170
name = f"{dataset_name}_{source}"
172171
graph.name = name
172+
graph.row_normalized = row_normalize
173173

174174
if verbosity:
175175
print_dataset_info(

0 commit comments

Comments
 (0)