Maximum Units on a Truck LeetCode solution in Python

1
class Solution:
2
    def maximumUnits(self, B: List[List[int]], T: int) -> int:
3
        B.sort(key=lambda x: x[1], reverse=True)
4
        ans = 0
5
        for b, n in B:
6
            boxes = min(b, T)
7
            ans += boxes * n
8
            T -= boxes
9
            if T == 0:
10
                return ans
11
        return ans