def METHOD_NAME(sagemaker_session):
    tf = _build_tf(
        sagemaker_session,
        framework_version="1.15.2",
        py_version="py3",
        environment=ENV_INPUT,
    )
    assert tf.environment == ENV_INPUT