
    nit                     v    S r SSKJr  SSKJr  SSKJr   SSKJs  Jr   " S S\5      rg! \	 a    SSK
r Nf = f)z7Mock algorithm for testing reinforcement learning code.    )absolute_import)division)print_functionNc                   6    \ rS rSrSrS rS rS rS rS r	Sr
g	)
MockAlgorithm   z+Produce random actions and empty summaries.c                     Xl         g)zZProduce random actions and empty summaries.

Args:
  envs: List of in-graph environments.
N_envs)selfenvss     c/home/james-whalen/.local/lib/python3.13/site-packages/pybullet_envs/agents/tools/mock_algorithm.py__init__MockAlgorithm.__init__   s	     J    c                 .    [         R                  " S5      $ N tfconstantr   unused_agent_indicess     r   begin_episodeMockAlgorithm.begin_episode$       ;;r?r   c                 v   [         R                  " U5      S   4U R                  S   R                  R                  -   nU R                  S   R                  R                  nU R                  S   R                  R
                  n[         R                  " U5      XT-
  -  U-   nU[         R                  " S5      4$ )Nr   r   )r   shaper   action_spacelowhighrandom_uniformr   )r   agent_indicesunused_observr   r    r!   actions          r   performMockAlgorithm.perform'   s    XXm$Q')DJJqM,F,F,L,LLE
**Q-
$
$
(
(C::a=%%**Du%4s:F2;;r?""r   c                 .    [         R                  " S5      $ r   r   )r   r   unused_transitions      r   
experienceMockAlgorithm.experience.   r   r   c                 .    [         R                  " S5      $ r   r   r   s     r   end_episodeMockAlgorithm.end_episode1   r   r   r
   N)__name__
__module____qualname____firstlineno____doc__r   r   r&   r*   r-   __static_attributes__ r   r   r   r      s    3#r   r   )r3   
__future__r   r   r   tensorflow.compat.v1compatv1r   	Exception
tensorflowobjectr   r5   r   r   <module>r=      s>    > &  %##
F 	  s   	+ 
88