1  import sys
 2  
 3  import mlflow
 4  
 5  
 6  def call_tracking_apis():
 7      mlflow.log_metric("some_key", 3)
 8  
 9  
10  def main(use_start_run):
11      if use_start_run:
12          with mlflow.start_run():
13              call_tracking_apis()
14      else:
15          call_tracking_apis()
16  
17  
18  if __name__ == "__main__":
19      main(use_start_run=int(sys.argv[1]))