code: def write_failure_file(failure_msg):  # type: (str) -> None
      
      file_path = os.path.join(_env.output_dir, 'failure')
      write_file(file_path, failure_msg)
docs: Create a file 'failure' if training fails after all algorithm output (for example, logging) completes,
      the failure description should be written to this file. In a DescribeTrainingJob response, Amazon SageMaker
      returns the first 1024 characters from this file as FailureReason.
      See: https://docs.aws.amazon.com/sagemaker/latest/dg/your-algorithms-training-algo.html
      Args:
          failure_msg: The description of failure