# -*- coding: utf-8 -*-
"""
cli.py
===
User-facing command-line functions for :module:`fto`.
"""
import string

from .fto import print_exercise, MassUnit


def process_input(units='lbs'):
    """Guide user through weight calculations via CLI prompts."""
    name = input("Please enter the exercise name: ")\
        .strip(string.whitespace)
    kgs = input("Kilograms? y or n: ").strip(string.whitespace)
    weight = int(input("Enter last week's max weight: "))
    week = int(input("Enter current training week: "))
    if week == 1:
        increment = int(input("How much are we adding? "))
    else:
        increment = 0
    units = MassUnit.kgs if kgs == 'y' else MassUnit.lbs

    print_exercise(name, weight, week, units, increment)


if __name__ == '__main__':
            process_input()

# -*- coding: utf-8 -*-
"""
cli.py
===
User-facing command-line functions for :module:`fto`.
"""
import string

from fto.fto import print_exercise, MassUnit


def process_input(units='lbs'):
    """Guide user through weight calculations via CLI prompts."""
    name = input("Please enter the exercise name: ")\
        .strip(string.whitespace)
    kgs = input("Kilograms? y or n: ").strip(string.whitespace)
    weight = int(input("Enter last week's max weight: "))
    week = int(input("Enter current training week: "))
    if week == 1:
        increment = int(input("How much are we adding? "))
    else:
        increment = 0
    units = MassUnit.kgs if kgs == 'y' else MassUnit.lbs

    print_exercise(name, weight, week, units, increment)


if __name__ == '__main__':
            process_input()

Use absolute import for python3
Use absolute import for python3
