Print all permutations of a given string in Python

string_permutations.py

'''
Write a program to print all the permutations of a given string. For example:
Input : ABC
Output: ABC ACB BAC BCA CAB CBA
'''

from itertools import permutations 

def print_permutations(s):
    """Prints permutations of a given string"""
    ans = list(permutations(s))
    print(s)
    for permutation in ans: 
        print(str().join(permutation), end = " ")
        
s = input('Enter input string\n')
print_permutations(s)