Function update_task_progress_file

examples/task-samples/task-byoo-sample/main.py:117–154

Creates a progress file to notify NVCT that a new output is ready to upload :param percent_complete: value between 0 and 100 to demonstrate percent complete :param progress_file_path: progress file path :param progress_name: name of folder/artifact to upload to NGC :param metad

(
    percent_complete: int,
    progress_file_path: str,
    progress_name: str,
    metadata: dict = None,
)

def update_task_progress_file(
    percent_complete: int,
    progress_file_path: str,
    progress_name: str,
    metadata: dict = None,
):
    """
    Creates a progress file to notify NVCT that a new output is ready to upload

    :param percent_complete: value between 0 and 100 to demonstrate percent complete
    :param progress_file_path: progress file path
    :param progress_name: name of folder/artifact to upload to NGC
    :param metadata: a dict of additional data to include
    :return:
    """
    # as requested in comment
    if metadata is None:
        metadata = dict()

    progress = Progress(
        taskId=os.getenv("NVCT_TASK_ID", ""),
        percentComplete=percent_complete,
        name=progress_name,
        lastUpdatedAt=datetime.datetime.now(datetime.timezone.utc).strftime(
            "%Y-%m-%dT%H:%M:%S.%fZ"
        ),
        metadata=metadata,
    )

    temp_file = progress_file_path + ".tmp"
    with open(temp_file, "w") as file:
        file.write(progress.output_to_json())

    try:
        os.rename(temp_file, progress_file_path)
    except Exception as e:
        logger.exception(f"Failed to write to progress file: {e}")
        raise e


def create_file_with_specific_size(filename, size_in_bytes):