/ tests / resources / example_project / greeter.py
greeter.py
 1  """
 2  Example program helping verify functionality for passing parameters other than those required in
 3  the MLproject file.
 4  """
 5  
 6  import argparse
 7  
 8  if __name__ == "__main__":
 9      parser = argparse.ArgumentParser()
10      parser.add_argument("greeting", help="Greeting to use", type=str)
11      parser.add_argument("name", help="Name of person to greet", type=str)
12      parser.add_argument("--excitement", help="Excitement level (int) of greeting", type=int)
13      args = parser.parse_args()
14      greeting = [args.greeting, args.name]
15      if args.excitement is not None:
16          greeting.append("!" * args.excitement)
17      print(" ".join(greeting))  # noqa: T201