def METHOD_NAME(sagemaker_local_session, docker_image, tmpdir, framework_version):
    output_path = "file://{}".format(tmpdir)
    run_tf_training(
        script=os.path.join(RESOURCE_PATH, "mnist", "mnist.py"),
        instance_type="local",
        instance_count=1,
        sagemaker_local_session=sagemaker_local_session,
        docker_image=docker_image,
        framework_version=framework_version,
        output_path=output_path,
        training_data_path="file://{}".format(os.path.join(RESOURCE_PATH, "mnist", "data")),
    )
    _assert_files_exist_in_tar(output_path, ["my_model.h5"])