o
    viZ
                     @   s   d Z ddlZddddZddddZdd	d	dZd
d Ze Zedkr&eZnedkr-eZnedkr3eZed Zed Z	ed Z
ede d ede  ede	  ede
  ed dS )a|  
Important constants for Cosmos Policy training and evaluation.

Attempts to automatically identify the correct constants to set based on the Python command used to launch
training or evaluation. If it is unclear, defaults to using the LIBERO simulation benchmark constants.

Adapted from: https://github.com/user/openvla-oft/blob/main/experiments/robot/libero/run_libero_eval.py
    N      	   )NUM_ACTIONS_CHUNK
ACTION_DIMPROPRIO_DIM    2      c                  C   s8   d tj } d| v rdS d| v rdS d| v rdS dS )N liberoLIBEROrobocasaROBOCASAalohaALOHA)joinsysargvlower)Zcmd_args r   =/data/cameron/vidgen/cosmos-policy/cosmos_policy/constants.pydetect_robot_platform0   s   r   r   r   r   r   r   r   zUsing z constants:z  NUM_ACTIONS_CHUNK = z  ACTION_DIM = z  PROPRIO_DIM = z^If needed, manually set the correct constants in `projects/cosmos/cosmos_policy/constants.py`!)__doc__r   ZLIBERO_CONSTANTSZROBOCASA_CONSTANTSZALOHA_CONSTANTSr   ZROBOT_PLATFORM	constantsr   r   r   printr   r   r   r   <module>   s<   	