Revert removing the unused imports (#385)

test/srt/model/reference_hf.py

@@ -1,4 +1,5 @@
 import argparse
+import os
 
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer

test/srt/model/test_llama_extend.py

@@ -1,6 +1,10 @@
 import multiprocessing
 import os
+import time
 
+import numpy as np
+import torch
+import torch.distributed as dist
 import transformers
 
 from sglang.srt.managers.router.infer_batch import Batch, ForwardMode, Req

test/srt/model/test_llava_low_api.py

@@ -1,10 +1,13 @@
 import multiprocessing
+import time
 
 import numpy as np
 import torch
+import torch.distributed as dist
 
 from sglang.srt.hf_transformers_utils import get_processor
-from sglang.srt.managers.router.model_runner import ModelRunner
+from sglang.srt.managers.router.infer_batch import ForwardMode
+from sglang.srt.managers.router.model_runner import InputMetadata, ModelRunner
 from sglang.srt.model_config import ModelConfig
 from sglang.srt.utils import load_image
 

test/srt/test_httpserver_concurrent.py

@@ -9,8 +9,11 @@ The capital of the United Kindom is London.\nThe capital of the United Kingdom i
 
 import argparse
 import asyncio
+import json
+import time
 
 import aiohttp
+import requests
 
 
 async def send_request(url, data, delay=0):

test/srt/test_httpserver_llava.py

@@ -10,6 +10,7 @@ The image features a man standing on the back of a yellow taxi cab, holding
 import argparse
 import asyncio
 import json
+import time
 
 import aiohttp
 import requests

test/srt/test_httpserver_reuse.py

@@ -6,6 +6,7 @@ The capital of France is Paris.\nThe capital of the United States is Washington,
 """
 
 import argparse
+import time
 
 import requests