Skip to content

Commit 20a75cb

Browse files
rename
Signed-off-by: Matthew Bonanni <mbonanni@redhat.com>
1 parent d85e431 commit 20a75cb

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vllm/attention/backends/registry.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ class _Backend(enum.Enum):
3131
TREE_ATTN = enum.auto()
3232

3333

34-
BACKEND_MAPPING = {}
34+
BACKEND_MAP = {}
3535

3636

3737
def register_attn_backend(backend: _Backend, class_path: str | None = None):
@@ -47,7 +47,7 @@ def register_attn_backend(backend: _Backend, class_path: str | None = None):
4747

4848
def decorator(cls):
4949
path = class_path or f"{cls.__module__}.{cls.__qualname__}"
50-
BACKEND_MAPPING[backend] = path
50+
BACKEND_MAP[backend] = path
5151
return cls
5252

5353
return decorator
@@ -62,7 +62,7 @@ def backend_to_class_str(backend: _Backend) -> str:
6262
Returns:
6363
The backend class string
6464
"""
65-
return BACKEND_MAPPING[backend]
65+
return BACKEND_MAP[backend]
6666

6767

6868
def backend_to_class(backend: _Backend) -> Type:

0 commit comments

Comments
 (0)