Update deprecated Python 3.8 typing (#13971)
This commit is contained in:
@@ -1,7 +1,5 @@
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
from typing import List
|
||||
|
||||
import pytest
|
||||
|
||||
from vllm.config import CompilationConfig, VllmConfig, set_current_vllm_config
|
||||
@@ -51,7 +49,7 @@ class Relu3(ReLUSquaredActivation):
|
||||
# All but RMSNorm
|
||||
("all,-rms_norm", 4, [0, 1, 1, 1], True),
|
||||
])
|
||||
def test_enabled_ops(env: str, torch_level: int, ops_enabled: List[int],
|
||||
def test_enabled_ops(env: str, torch_level: int, ops_enabled: list[int],
|
||||
default_on: bool):
|
||||
vllm_config = VllmConfig(compilation_config=CompilationConfig(
|
||||
level=torch_level, custom_ops=env.split(",")))
|
||||
|
||||
Reference in New Issue
Block a user