
    h                         S r SSKrSSKJr  SSKJr  SSKJrJr  SSK	J
r
  SSKJr  SSKJrJrJrJr  SS	KJrJrJrJrJrJrJrJrJr  SS
KJrJrJr  / SQr SSSSS.r!S\"4S jr#g)z!Wrappers for vector environments.    N)RecordEpisodeStatistics)DictInfoToList)HumanRenderingRecordVideo)NormalizeObservation)NormalizeReward)
ClipActionRescaleActionTransformActionVectorizeTransformAction)	DtypeObservationFilterObservationFlattenObservationGrayscaleObservationRescaleObservationReshapeObservationResizeObservationTransformObservationVectorizeTransformObservation)
ClipRewardTransformRewardVectorizeTransformReward)r   r   r   r   r   r   r   r   r   r   r   r   r   r   r	   r
   r   r   r   r   r   r   ArrayConversion
JaxToNumpy
JaxToTorchNumpyToTorcharray_conversionjax_to_numpyjax_to_torchnumpy_to_torch)r   r   r   r   wrapper_namec                     U [         ;   a-  S[         U     3n[        R                  " U5      n[        X 5      $ [	        S[
        < SU < 35      e)a  Load a wrapper by name.

This optimizes the loading of gymnasium wrappers by only loading the wrapper if it is used.
Errors will be raised if the wrapper does not exist or if the version is not the latest.

Args:
    wrapper_name: The name of a wrapper to load.

Returns:
    The specified wrapper.

Raises:
    AttributeError: If the wrapper does not exist.
    DeprecatedWrapper: If the version is not the latest.
zgymnasium.wrappers.vector.zmodule z has no attribute )_wrapper_to_class	importlibimport_modulegetattrAttributeError__name__)r!   import_stmtmodules      \/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/wrappers/vector/__init__.py__getattr__r,   X   sU    " ((23D\3R2ST((5v,,
78,.@@PQ
RR    )$__doc__r$    gymnasium.wrappers.vector.commonr   +gymnasium.wrappers.vector.dict_info_to_listr   #gymnasium.wrappers.vector.renderingr   r   .gymnasium.wrappers.vector.stateful_observationr   )gymnasium.wrappers.vector.stateful_rewardr   *gymnasium.wrappers.vector.vectorize_actionr	   r
   r   r   /gymnasium.wrappers.vector.vectorize_observationr   r   r   r   r   r   r   r   r   *gymnasium.wrappers.vector.vectorize_rewardr   r   r   __all__r#   strr,    r-   r+   <module>r:      sk    '  D F K O E 
 
 
 '\ *  $ Sc Sr-   