Skip to content

Add describe_transform_job() to the SageMaker Session class #1502

Closed
@alex23lemm

Description

@alex23lemm

Is your feature request related to a problem? Please describe.
A SageMaker Session object comes with the following two methods which allow me to retrieve the status of training and AutoML jobs: describe_training_job(), describe_auto_ml_job(). Example:

import sagemaker

session = sagemaker.Session()
session.describe_training_job("[YOUR_JOB_NAME]")["TrainingJobStatus"]

However, if I like to get the status of a batch transform job (like, e.g., a batch interence job) in the same workflow, I need to create an extra low-level SageMaker client using boto3:

import boto3

sm = boto3.client("sagemaker")
sm.describe_transform_job(TransformJobName = "[YOUR_JOB_NAME]")["TransformJobStatus"]

Describe the solution you'd like
Even though the above workflow works it could become much more concise and smoother by adding a describe_transform_job() method to the SageMaker Session class. This would eliminate the need to use the extra low-level SageMaker client in the same workflow.

Metadata

Metadata

Assignees

No one assigned

    Labels

    status: pending releaseThe fix have been merged but not yet released to PyPI

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions