diff --git a/tests/test_booster/test_mixed_precision/test_fp16_torch.py b/tests/test_booster/test_mixed_precision/test_fp16_torch.py index d98171a3d..bb76b354d 100644 --- a/tests/test_booster/test_mixed_precision/test_fp16_torch.py +++ b/tests/test_booster/test_mixed_precision/test_fp16_torch.py @@ -1,6 +1,6 @@ +import pytest import torch from torch.optim import Adam -import pytest import colossalai from colossalai.booster.mixed_precision import FP16TorchMixedPrecision