Skip to content

GitLab

Unverified Commit 4726c919 authored by Liangsheng Yin's avatar Liangsheng Yin Committed by GitHub
Browse files

[minor] fix the lint (#11198)

parent a0010bf4
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 2 deletions
test/srt/models/test_vlm_models.py
View file @ 4726c919
...@@ -27,9 +27,10 @@ MODELS = [ ...@@ -27,9 +27,10 @@ MODELS = [
SimpleNamespace(model="openbmb/MiniCPM-V-2_6", mmmu_accuracy=0.4), SimpleNamespace(model="openbmb/MiniCPM-V-2_6", mmmu_accuracy=0.4),
] ]
#Set default mem_fraction_static to 0.8 # Set default mem_fraction_static to 0.8
DEFAULT_MEM_FRACTION_STATIC = 0.8 DEFAULT_MEM_FRACTION_STATIC = 0.8
class TestVLMModels(CustomTestCase): class TestVLMModels(CustomTestCase):
parsed_args = None # Class variable to store args parsed_args = None # Class variable to store args
...@@ -41,7 +42,9 @@ class TestVLMModels(CustomTestCase): ...@@ -41,7 +42,9 @@ class TestVLMModels(CustomTestCase):
cls.time_out = DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH cls.time_out = DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH
if cls.parsed_args is None: if cls.parsed_args is None:
cls.parsed_args = SimpleNamespace(mem_fraction_static=DEFAULT_MEM_FRACTION_STATIC) cls.parsed_args = SimpleNamespace(
mem_fraction_static=DEFAULT_MEM_FRACTION_STATIC
)
# Set OpenAI API key and base URL environment variables. Needed for lmm-evals to work. # Set OpenAI API key and base URL environment variables. Needed for lmm-evals to work.
os.environ["OPENAI_API_KEY"] = cls.api_key os.environ["OPENAI_API_KEY"] = cls.api_key
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment