1 Преобразование числа, введенного пользователем, в байты или кб, в зависимости от выбора числа. ​

задание , которое на русском:

def main():

   n = int(input())

   t = input()

   if t == "k":

       print("{} kb, {} b".format(n, n * 1024))

   elif t == "b":

       print("{} kb, {} b".format(n / 1024, n))

   else:

       raise KeyError("Type of value must be 'k' or 'b'")

if __name__ == "__main__":

   main()