Skip to content

Commit cece6de

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 3e34f79 commit cece6de

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

torch_geometric/datasets/web_qsp_dataset.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,12 @@
22
import gc
33
import os
44
from itertools import chain
5-
from typing import Any, Dict, Iterator, List, Tuple, no_type_check
5+
from typing import Any, Dict, Iterator, List
66

7-
import numpy as np
87
import torch
9-
from torch import Tensor
108
from tqdm import tqdm
119

1210
from torch_geometric.data import (
13-
Data,
1411
InMemoryDataset,
1512
LargeGraphIndexer,
1613
TripletLike,

torch_geometric/utils/rag/backend_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ def preprocess_triplet(triplet: TripletLike) -> TripletLike:
4949
h, r, t = triplet
5050
return str(h).lower(), str(r).lower(), str(t).lower()
5151

52+
5253
@no_type_check
5354
def retrieval_via_pcst(
5455
data: Data,
@@ -173,6 +174,7 @@ def retrieval_via_pcst(
173174

174175
return data, desc
175176

177+
176178
def batch_knn(query_enc: Tensor, embeds: Tensor,
177179
k: int) -> Iterator[InputNodes]:
178180
from torchmetrics.functional import pairwise_cosine_similarity

0 commit comments

Comments
 (0)