Fix relative import

hpcaitech · Nov 13, 2024 · 7f744a0 · 7f744a0
1 parent 98fa553
commit 7f744a0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_optimizer/test_adam_optim.py b/tests/test_optimizer/test_adam_optim.py
@@ -8,8 +8,7 @@
 
 from colossalai.nn.optimizer import CPUAdam, FusedAdam, HybridAdam
 from tests.kit.model_zoo import model_zoo
-
-from ._utils import force_assign_grad, setup_param_groups
+from tests.test_optimizer._utils import force_assign_grad, setup_param_groups
 
 _ALLOWED_OPTIM_DEVICES = [
     (FusedAdam, torch.device("cuda:0")),