caikit.runtime.servicers.training_management_servicer
=====================================================

.. py:module:: caikit.runtime.servicers.training_management_servicer


Attributes
----------

.. autoapisummary::

   caikit.runtime.servicers.training_management_servicer.log


Classes
-------

.. autoapisummary::

   caikit.runtime.servicers.training_management_servicer.TrainingManagementServicerImpl


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

.. py:data:: log

.. py:class:: TrainingManagementServicerImpl

   This class contains the implementation of all of the RPCs that are required to run a
   service in Model Mesh as a Model-Runtime.


   .. py:method:: GetTrainingStatus(request, context)

      Get the status of a training by ID



   .. py:method:: CancelTraining(request, context)

      Cancel a training future.



   .. py:method:: get_training_status(training_id: str) -> caikit.interfaces.runtime.data_model.TrainingStatusResponse

      Get the status of a training by ID



   .. py:method:: cancel_training(training_id: str) -> caikit.interfaces.runtime.data_model.TrainingStatusResponse

      Cancel a training future.



   .. py:method:: _get_model_future(training_id: str, operation: str)
      :staticmethod:


      Returns a model future, or raises 404 caikit runtime exception on error.
      Wrapped here so that we only catch errors directly in the `trainer.get_model_future` call