diff --git a/tests/random_test/random_test.py b/tests/random_test/random_test.py index 32135a1..534312a 100644 --- a/tests/random_test/random_test.py +++ b/tests/random_test/random_test.py @@ -6,7 +6,12 @@ from torch import nn from torch.utils.data import Dataset -from easytorch import Config, Runner, get_rank, launch_training +#path patch +import sys +sys.path.append('../..') + +from easytorch import Config, Runner, launch_training +from easytorch.utils import get_rank class FakeDataset(Dataset):