diff --git a/examples/llm/g_retriever.py b/examples/llm/g_retriever.py index 0acf52ece352..5b38b457fdfb 100644 --- a/examples/llm/g_retriever.py +++ b/examples/llm/g_retriever.py @@ -16,7 +16,6 @@ import os.path as osp import re import time -from torch_geometric import seed_everything import pandas as pd import torch @@ -24,6 +23,7 @@ from torch.nn.utils import clip_grad_norm_ from tqdm import tqdm +from torch_geometric import seed_everything from torch_geometric.datasets import WebQSPDataset from torch_geometric.loader import DataLoader from torch_geometric.nn.models import GAT, GRetriever