!!!! Помогите пожалуйста с Питоном. Задача такая: написать программу, в которой вводятся 2 числа и если они равны, то распечатать число, если не равны, то распечатать их сумму

a = int(input())

b = int(input())

if a == b:

   print(a)

else:

   print(a+b)

Ну или если перед пацанами выпендриться хочется:

a,b=int(input()),int(input())

print(a if a==b else a+b)