triton_wrapper
==============

.. currentmodule:: coffea.ml_tools

.. autoclass:: triton_wrapper
   :show-inheritance:

   .. rubric:: Attributes Summary

   .. autosummary::

      ~triton_wrapper.batch_size
      ~triton_wrapper.batch_size_fallback
      ~triton_wrapper.client_args
      ~triton_wrapper.http_client_concurrency
      ~triton_wrapper.max_retry_attempts
      ~triton_wrapper.pmod
      ~triton_wrapper.retry_jitter_base_ms

   .. rubric:: Methods Summary

   .. autosummary::

      ~triton_wrapper.numpy_call
      ~triton_wrapper.run_infer
      ~triton_wrapper.validate_numpy_input

   .. rubric:: Attributes Documentation

   .. autoattribute:: batch_size
   .. autoattribute:: batch_size_fallback
   .. autoattribute:: client_args
   .. autoattribute:: http_client_concurrency
   .. autoattribute:: max_retry_attempts
   .. autoattribute:: pmod
   .. autoattribute:: retry_jitter_base_ms

   .. rubric:: Methods Documentation

   .. automethod:: numpy_call
   .. automethod:: run_infer
   .. automethod:: validate_numpy_input