frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-26 00:24:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE]: Simplify some list comps to generators C419 (#132578)

Browse Source 
Simplifies some list comprehensions to generator which is more efficient. Automatically applied diffs for the most part with ruff

Pull Request resolved: https://github.com/pytorch/pytorch/pull/132578
Approved by: https://github.com/ezyang
This commit is contained in:
Aaron Gokaslan
2024-08-04 17:46:24 +00:00
committed by PyTorch MergeBot
parent 4226ed1585
commit fd4b649e6c
7 changed files with 26 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/nested/_internal/nested_tensor.py
View File

@ -414,8 +414,8 @@ def jagged_from_list(
)
# compute this now since it's easy
min_seqlen = min([t.shape[0] for t in tensors])
max_seqlen = max([t.shape[0] for t in tensors])
min_seqlen = min(t.shape[0] for t in tensors)
max_seqlen = max(t.shape[0] for t in tensors)
ret_nt = nested_view_from_values_offsets(
values, offsets, min_seqlen=min_seqlen, max_seqlen=max_seqlen
)