demo_test.py

import os
import argparse

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'


def fun(x):
    return x + 1


if __name__ == '__main__':
    pa = argparse.ArgumentParser(description='tensorflow constants and functions')
    pa.add_argument('--operation', dest='operation', required=False, help='operation')
    # constant, square, transpose
    args = pa.parse_args()