Skip to content

Commit fe1501b

Browse files
committed
Fix checkstyle
1 parent 034d826 commit fe1501b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

graphdatascience/graph/ogb_loader.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def _parse_homogeneous(self, dataset: HomogeneousOGBNDataset) -> tuple[list[pd.D
111111
node_dict["features"] = graph["node_feat"].tolist() # type: ignore
112112

113113
if len(dataset.labels[0]) == 1:
114-
node_dict["classLabel"] = [cl[0] for cl in dataset.labels]
114+
node_dict["classLabel"] = [cl[0] for cl in dataset.labels] # type: ignore
115115
else:
116116
node_dict["classLabel"] = dataset.labels.tolist() # type: ignore
117117

@@ -174,7 +174,7 @@ def _parse_heterogeneous(self, dataset: HeterogeneousOGBNDataset) -> tuple[list[
174174

175175
if node_label in class_labels:
176176
if len(class_labels[node_label]) == 1:
177-
node_dict["classLabel"] = [cl[0] for cl in class_labels[node_label]]
177+
node_dict["classLabel"] = [cl[0] for cl in class_labels[node_label]] # type: ignore
178178
else:
179179
node_dict["classLabel"] = class_labels[node_label].tolist() # type: ignore
180180

0 commit comments

Comments
 (0)