206447f53847129741d0b1abff789b2788e89a3b,solutionbox/structured_data/datalab_structured_data/_package.py,,local_train,#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,178

Before Change

  stderr = sys.stderr
  sys.stderr = sys.stdout
  print("Starting local training.")
  trainer.task.main(args)
  print("Local training done.")
  sys.stderr = stderr

def cloud_train(train_dataset,

After Change

  monitor_process = None
  try:
    print("Starting local training")
    p = subprocess.Popen(" ".join(args),
                         shell=True,
                         stdout=subprocess.PIPE,
                         stderr=subprocess.STDOUT)
    pids_to_kill = [p.pid]

    //script -> name = datalab_structured_data._package
    script = "import %s; %s._wait_and_kill(%s, %s)" % \
          (__name__, __name__, str(os.getpid()), str(pids_to_kill))
    monitor_process = subprocess.Popen(["python", "-c", script])

    while p.poll() is None:
      sys.stdout.write(p.stdout.readline())
    print("Local training done.")

Instances