diff --git a/sorts/cyclesort.py b/sorts/cyclesort.py index d6762b083..ee19a1ade 100644 --- a/sorts/cyclesort.py +++ b/sorts/cyclesort.py @@ -44,7 +44,13 @@ def cycle_sort(array): # Main Code starts here -user_input = input('Enter numbers separated by a comma:\n') +if __name__ == '__main__': + try: + raw_input # Python 2 + except NameError: + raw_input = input # Python 3 + +user_input = raw_input('Enter numbers separated by a comma:\n') unsorted = [int(item) for item in user_input.split(',')] n = len(unsorted) cycle_sort(unsorted)