feluelle commented on a change in pull request #9594:
URL: https://github.com/apache/airflow/pull/9594#discussion_r460809299
##########
File path: airflow/providers/amazon/aws/hooks/sagemaker.py
##########
@@ -786,6 +825,30 @@ def list_training_jobs(
)
return results
+ def list_processing_jobs(self, **kwargs) -> List[Dict]: # noqa: D402
+ """
+ This method wraps boto3's list_processing_jobs(). All arguments should
be provided via kwargs.
+ Note boto3 expects these in CamelCase format, for example:
+
+ .. code-block:: python
+
+ list_processing_jobs(NameContains="myjob", StatusEquals="Failed")
+
+ .. seealso::
+
https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.list_processing_jobs
+
+ :param kwargs: (optional) kwargs to boto3's list_training_jobs method
+ :return: results of the list_processing_jobs request
+ """
+
+ config = {}
+ config.update(kwargs)
+ list_processing_jobs_request =
partial(self.get_conn().list_processing_jobs, **config)
Review comment:
```suggestion
list_processing_jobs_request =
partial(self.get_conn().list_processing_jobs, **kwargs)
```
Correct?
----------------------------------------------------------------
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.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
