Unverified Commit 0528e90d authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[Examples] fix path for load_graph (#3797)

parent 1d80d91a
......@@ -11,7 +11,7 @@ import tqdm
import glob
import os
import sys
sys.path.append('../')
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
from load_graph import load_reddit, inductive_split, load_ogb
from torchmetrics import Accuracy
......
......@@ -18,7 +18,7 @@ from pytorch_lightning.callbacks import ModelCheckpoint, Callback
from pytorch_lightning import LightningDataModule, LightningModule, Trainer
from model import SAGE, compute_acc_unsupervised as compute_acc
import sys
sys.path.append('../')
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
from load_graph import load_reddit, inductive_split, load_ogb
class CrossEntropyLoss(nn.Module):
......
......@@ -15,7 +15,7 @@ from torch.nn.parallel import DistributedDataParallel
from model import SAGE, compute_acc_unsupervised as compute_acc
from negative_sampler import NegativeSampler
import sys
sys.path.append('../')
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
from load_graph import load_reddit, load_ogb
class CrossEntropyLoss(nn.Module):
......
......@@ -4,7 +4,8 @@ import torch as th
import argparse
import time
import sys
sys.path.append('../')
import os
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
from load_graph import load_reddit, load_ogb
if __name__ == '__main__':
......
......@@ -19,7 +19,7 @@ from statistics import mean
import random
import time
import argparse
sys.path.append('../')
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
from load_graph import load_ogb
import dgl
from dgl.data import load_data
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment