o
    i+                     @   sJ   d dl Z d dlT erddlmZ ddlmZ dede	e
d  fdd	ZdS )
    N)*   )	MoGeModelversionreturn)MoGeModelV1MoGeModelV2c                 C   sZ   | dv sJ d|  zt d|  t}W n ty%   td|  dw t|d}|S )N)v1v2zUnsupported model version: .zModel version "z" not found.r   )	importlibimport_module__package__ModuleNotFoundError
ValueErrorgetattr)r   modulecls r   ./data/cameron/moge_repo/moge/model/__init__.pyimport_model_class_by_version	   s   
r   )r   typingZTYPE_CHECKINGr	   r   r   r
   r   strZTypeZUnionr   r   r   r   r   <module>   s    