From 3ff366d2f0b9007fd2d85f4bc8bfa01f4d56872f Mon Sep 17 00:00:00 2001
From: deanmongel <5518794+deanmongel@users.noreply.github.com>
Date: Fri, 28 Feb 2025 15:36:41 +0800
Subject: [PATCH] fix: fix path&import bug in random_test

---
 tests/random_test/random_test.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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):