Skip to content

Commit 3425cd5

Browse files
authored
docs(mta): suppress invalid sequence syntax warning (#870)
## Summary <!--- This is a required section; please describe the main purpose of this proposed code change. ---> According to [PEP-257](https://peps.python.org/pep-0257/): > For consistency, always use """triple double quotes""" around docstrings. Use r"""raw triple double quotes""" if you use any backslashes in your docstrings. Fix #866 and a coding style issue in setup.py <!--- ## Details This is an optional section; is there anything specific that reviewers should be aware of? ---> ## Testing Done <!--- This is a required section; please describe how this change was tested. ---> <!-- Replace BLANK with your device type. For example, A100-80G-PCIe Complete the following tasks before sending your PR, and replace `[ ]` with `[x]` to indicate you have done them. --> - Hardware Type: <BLANK> - [ ] run `make test` to ensure correctness - [x] run `make checkstyle` to ensure code style - [ ] run `make test-convergence` to ensure convergence --------- Signed-off-by: Tcc0403 <[email protected]>
1 parent d9107b7 commit 3425cd5

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,5 +109,5 @@ def get_platform() -> Literal["cuda", "rocm", "cpu", "xpu"]:
109109
"Topic :: Software Development :: Libraries :: Python Modules",
110110
"License :: OSI Approved :: BSD-2-Clause Software License",
111111
"Operating System :: OS Independent",
112-
]
112+
],
113113
)

src/liger_kernel/transformers/multi_token_attention.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010

1111
class LigerMultiTokenAttention(nn.Module):
12-
"""
12+
r"""
1313
Multi-Token Attention:
1414
out = mask_{0}(conv2d(softmax(mask_{-\inf}(scores))))
1515

0 commit comments

Comments
 (0)