o
    ci                     @   s   d Z ddlmZ ddlmZ eddddZe Ze jdej	p!d d	j
ejd
ddddjejp3ddej	p8d ddjej	pAdddddjdddej	pNd dZedkreddlmZ eeedd dS dS )a  
[1] Mastering Diverse Domains through World Models - 2023
D. Hafner, J. Pasukonis, J. Ba, T. Lillicrap
https://arxiv.org/pdf/2301.04104v1.pdf

[2] Mastering Atari with Discrete World Models - 2021
D. Hafner, T. Lillicrap, M. Norouzi, J. Ba
https://arxiv.org/pdf/2010.02193.pdf
    )DreamerV3Config)add_rllib_example_script_argsi@B g      4@)default_itersdefault_rewarddefault_timesteps      )num_cpus_for_main_processg        F)repeat_action_probabilityfull_action_space	frameskip)env
env_configT)num_env_runnersnum_envs_per_env_runnerremote_worker_envs)"metrics_num_episodes_for_smoothingreport_images_and_videosreport_dream_data"report_individual_batch_item_statsXL@      )
model_sizetraining_ratiobatch_size_B__main__)#run_rllib_example_script_experiment)keep_configN)__doc__(ray.rllib.algorithms.dreamerv3.dreamerv3r   ray.rllib.utils.test_utilsr   parser
parse_argsargs	resourcesnum_learnersenvironmentr   env_runnersr   	reportingtrainingconfig__name__r    r-   r-   a/home/ubuntu/.local/lib/python3.10/site-packages/ray/rllib/tuned_examples/dreamerv3/atari_200M.py<module>   sV    
 '0