
    HhU                         d dl mZ ddlmZ ddlmZ ddlmZ ddlmZ ddl	mZ
 ddl	mZ  ed	           G d
 d                      Z e            Zd ZdS )    )	dataclass   )get_encoder)AlignedShapeLatentPerceiver)get_encoder_simplified)ShapeAsLatentPerceiverEncoder)PointTransformerV3ObjectT)frozenc                       e Zd ZeZeZeZdS )_MAP_MESH_ENCODERN)	__name__
__module____qualname__r	   ptv3objr   michelangelor   michelangelo_encoder     H/data/cameron/tmprepos/clean_dreammesh/UniRig/src/model/parse_encoder.pyr   r   
   s         &G.L8r   r   c                      t           t          t          d}| d         }| d= ||v s5J dd                    |                                           d|              ||         di | S )N)r   r   r   
__target__z	expect: [,z
], found: r   )get_encoder_ptv3objget_encoder_michelangelo get_encoder_michelangelo_encoderjoinkeys)kwargsMAPr   s      r   get_mesh_encoderr       s    &0 @ C
 %J|V#((388::*>*>VV*VV3z?$$V$$$r   N)dataclassesr   michelangelo.get_modelr   r   r   r   r   r   pointcept.models.PTv3Objectr   r	   r   MAP_MESH_ENCODERr    r   r   r   <module>r%      s    ! ! ! ! ! ! K K K K K K ? ? ? ? ? ? ^ ^ ^ ^ ^ ^ A A A A A A K K K K K K A A A A A A
$9 9 9 9 9 9 9 9
 %$&& 	% 	% 	% 	% 	%r   