Third Maximum Number

Given an integer array nums, return the third distinct maximum number in this array. If the third maximum does not exist, return the maximum number.

class Solution:
    def thirdMax(self, nums: List[int]) -> int:
        n = list(set(nums))

        if len(n) < 3:
            return max(n)

        return sorted(n)[-3]