hsrocks commented on a change in pull request #21673:
URL: https://github.com/apache/airflow/pull/21673#discussion_r814029742
##########
File path: airflow/providers/amazon/aws/operators/sagemaker.py
##########
@@ -608,3 +608,26 @@ def _check_if_job_exists(self) -> None:
raise AirflowException(
f'A SageMaker training job with name {training_job_name}
already exists.'
)
+
+
+class SageMakerDeleteModelOperator(BaseOperator):
Review comment:
Done the changes. I implemented BaseOperator because with
SagemakerBaseOperator , we have to pass config dictionary but delete model call
just need ModelName as argument but to ensure consistency with other Sagemaker
Operator I have made necessary. Please check and resolve. Thanks
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
