diff --git a/tests/st/gnn/gcn/test_gcn.py b/tests/st/gnn/gcn/test_gcn.py index 8ab1f3a6572febd0fa814cbcdea6532068129860..09789a00e576a6f90ab61ad98bed126d0542666e 100644 --- a/tests/st/gnn/gcn/test_gcn.py +++ b/tests/st/gnn/gcn/test_gcn.py @@ -17,10 +17,10 @@ import time import pytest import numpy as np from mindspore import context -from model_zoo.gnn.gcn.src.gcn import GCN -from model_zoo.gnn.gcn.src.metrics import LossAccuracyWrapper, TrainNetWrapper -from model_zoo.gnn.gcn.src.config import ConfigGCN -from model_zoo.gnn.gcn.src.dataset import get_adj_features_labels, get_mask +from model_zoo.official.gnn.gcn.src.gcn import GCN +from model_zoo.official.gnn.gcn.src.metrics import LossAccuracyWrapper, TrainNetWrapper +from model_zoo.official.gnn.gcn.src.config import ConfigGCN +from model_zoo.official.gnn.gcn.src.dataset import get_adj_features_labels, get_mask DATA_DIR = '/home/workspace/mindspore_dataset/cora/cora_mr/cora_mr'