Merge pull request #159 from kellyaa/main

add .complete file to output dir when done

build/launch_training.py

@@ -18,6 +18,7 @@
 """
 
 # Standard
+from pathlib import Path
 import os
 import tempfile
 import shutil
@@ -187,6 +188,9 @@ def main():
  )
  if os.path.exists(train_logs_filepath):
  shutil.copy(train_logs_filepath, original_output_dir)
+ # The .complete file will signal to users that we are finished copying
+ # files over
+ Path(os.path.join(original_output_dir, ".complete")).touch()
  except Exception as e: # pylint: disable=broad-except
  logging.error(traceback.format_exc())
  write_termination_log(

tests/build/test_launch_script.py

@@ -88,6 +88,7 @@ def test_successful_pt():
 
  assert main() == 0
  assert os.path.exists(tempdir + "/termination-log") is False
+ assert os.path.exists(os.path.join(tempdir, ".complete")) is True
 
 
 def test_bad_script_path():