etha.tensor_bus.agent
=====================

.. py:module:: etha.tensor_bus.agent

.. autoapi-nested-parse::

   Tensor Bus Agent Process.



Attributes
----------

.. autoapisummary::

   etha.tensor_bus.agent.TIME_INTERVAL
   etha.tensor_bus.agent.logfire
   etha.tensor_bus.agent.logger


Classes
-------

.. autoapisummary::

   etha.tensor_bus.agent.TensorBusAgent


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

.. py:class:: TensorBusAgent(rank: int, world_size: int, store_host: str, store_port: int, lmdb_command_queue_path: str, lmdb_state_path: str, store_timeout: float = 3600.0, store_backend: str = 'tcp', store_namespace: str | None = None)

   Tensor Bus Agent.

   Initialize Agent.

   :param rank: Rank in the torch.distributed group
   :param world_size: Total number of Agents
   :param store_host: KVStore server host
   :param store_port: KVStore server port
   :param lmdb_command_queue_path: Path to CommandQueue LMDB
   :param lmdb_state_path: Path to State LMDB
   :param store_timeout: KVStore connection timeout in seconds
   :param store_backend: KVStore backend ("tcp" or "etcd")


   .. py:method:: close(destroy: bool = True)

      Cleanup resources.



   .. py:method:: run()

      Main loop: process commands from Host.



   .. py:attribute:: batches
      :type:  dict[str, etha.tensor_bus.batch_state.BatchState]


   .. py:attribute:: command_queue


   .. py:attribute:: lmdb_state_path


   .. py:attribute:: pairs
      :type:  dict[str, etha.tensor_bus.pair_state.PairState]


   .. py:attribute:: rank


   .. py:attribute:: state_db


   .. py:attribute:: state_env


   .. py:attribute:: store
      :type:  etha.kvstore.KVStore


   .. py:attribute:: world_size


.. py:data:: TIME_INTERVAL
   :value: 0.001


.. py:data:: logfire
   :value: None


.. py:data:: logger