o
    `۷i                     @   sR   d dl Z d dlmZmZ d dlmZmZmZ d dlm	Z	 e	G dd dedZ
dS )    N)ABCMetaabstractmethod)AnyDictOptional)DeveloperAPIc                   @   s   e Zd ZdZeedefddZeededdfddZ	eedd	e
e defd
dZeedeeef fddZeedeeef ddfddZedefddZdS )ReplayBufferInterfacea  Abstract base class for all of RLlib's replay buffers.

    Mainly defines the `add()` and `sample()` methods that every buffer class
    must implement to be usable by an Algorithm.
    Buffers may determine on all the implementation details themselves, e.g.
    whether to store single timesteps, episodes, or episode fragments or whether
    to return fixed batch sizes or per-call defined ones.
    returnc                 C      dS )z<Returns the number of items currently stored in this buffer.N selfr   r   Y/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/rllib/utils/replay_buffers/base.py__len__       zReplayBufferInterface.__len__batchNc                 K   r
   )zAdds a batch of experiences or other data to this buffer.

        Args:
            batch: Batch or data to add.
            ``**kwargs``: Forward compatibility kwargs.
        Nr   )r   r   kwargsr   r   r   add   r   zReplayBufferInterface.add	num_itemsc                 K   r
   )aD  Samples `num_items` items from this buffer.

        The exact shape of the returned data depends on the buffer's implementation.

        Args:
            num_items: Number of items to sample from this buffer.
            ``**kwargs``: Forward compatibility kwargs.

        Returns:
            A batch of items.
        Nr   )r   r   r   r   r   r   sample"   r   zReplayBufferInterface.samplec                 C   r
   )zgReturns all local state in a dict.

        Returns:
            The serializable local state.
        Nr   r   r   r   r   	get_state1   r   zReplayBufferInterface.get_statestatec                 C   r
   )zRestores all local state to the provided `state`.

        Args:
            state: The new state to set this buffer. Can be obtained by calling
                `self.get_state()`.
        Nr   )r   r   r   r   r   	set_state:   r   zReplayBufferInterface.set_statec                 C   s   t  S )zReturns the computer's network name.

        Returns:
            The computer's networks name or an empty string, if the network
            name could not be determined.
        )platformnoder   r   r   r   get_hostD   s   zReplayBufferInterface.get_host)N)__name__
__module____qualname____doc__r   r   intr   r   r   r   r   r   strr   r   r   r   r   r   r   r      s&    	r   )	metaclass)r   abcr   r   typingr   r   r   ray.util.annotationsr   r   r   r   r   r   <module>   s    