:py:mod:`astronomer.providers.amazon.aws.hooks.sagemaker`
=========================================================

.. py:module:: astronomer.providers.amazon.aws.hooks.sagemaker


Module Contents
---------------

Classes
~~~~~~~

.. autoapisummary::

   astronomer.providers.amazon.aws.hooks.sagemaker.SageMakerHookAsync




.. py:class:: SageMakerHookAsync(*args, **kwargs)

   Bases: :py:obj:`astronomer.providers.amazon.aws.hooks.base_aws.AwsBaseHookAsync`

   Interact with Amazon SageMaker async using aiobotocore python library.

   Additional arguments (such as ``aws_conn_id``) may be specified and
   are passed down to the underlying AwsBaseHookAsync.

   .. py:attribute:: NON_TERMINAL_STATES
      :value: ('InProgress', 'Stopping', 'Stopped')

      

   .. py:method:: describe_transform_job_async(job_name)
      :async:

      Return the transform job info associated with the name

      :param job_name: the name of the transform job


   .. py:method:: describe_processing_job_async(job_name)
      :async:

      Return the processing job info associated with the name

      :param job_name: the name of the processing job


   .. py:method:: describe_training_job_async(job_name)
      :async:

      Return the training job info associated with the name

      :param job_name: the name of the training job


   .. py:method:: describe_training_job_with_log(job_name, positions, stream_names, instance_count, state, last_description, last_describe_job_call)
      :async:

      Return the training job info associated with job_name and print CloudWatch logs

      :param job_name: name of the job to check status
      :param positions: A list of pairs of (timestamp, skip) which represents the last record
          read from each stream.
      :param stream_names: A list of the log stream names. The position of the stream in this list is
          the stream number.
      :param instance_count: Count of the instance created for the job initially
      :param state: log state
      :param last_description: Latest description of the training job
      :param last_describe_job_call: previous job called time


   .. py:method:: get_multi_stream(log_group, streams, positions)
      :async:

      Iterate over the available events coming from a set of log streams in a single log group
      interleaving the events from each stream so they're yielded in timestamp order.

      :param log_group: The name of the log group.
      :param streams: A list of the log stream names. The position of the stream in this list is
          the stream number.
      :param positions: A list of pairs of (timestamp, skip) which represents the last record
          read from each stream.