From cd371d31fc7d62d6d28b5a803a31a3a1accc3d35 Mon Sep 17 00:00:00 2001
From: Shangyan Zhou <sy.zhou@deepseek.com>
Date: Wed, 18 Jun 2025 14:52:04 +0800
Subject: [PATCH] Move import.

---
 tests/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/utils.py b/tests/utils.py
index 7af4947..57a38b2 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -4,6 +4,7 @@ import numpy as np
 import torch
 import torch.distributed as dist
 from typing import Optional
+import inspect
 
 
 def init_dist(local_rank: int, num_local_ranks: int):
@@ -14,7 +15,6 @@ def init_dist(local_rank: int, num_local_ranks: int):
     node_rank = int(os.getenv('RANK', 0))
     assert (num_local_ranks < 8 and num_nodes == 1) or num_local_ranks == 8
 
-    import inspect
     sig = inspect.signature(dist.init_process_group)
     params = {
         'backend': 'nccl',