name: tensorflow_linear_regression_example python_env: python_env.yaml entry_points: main: command: "python train.py"