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ej	dd	d
ddj
ejp*dejp.d
ejd
kdjejp9d
d	d	d	djdddejpFd
 dZedkr]d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@i )default_itersdefault_rewarddefault_timestepsg        F   )repeat_action_probabilityfull_action_space	frameskip)env
env_config)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Si      )
model_sizetraining_ratiobatch_size_B__main__)#run_rllib_example_script_experimentT)keep_configN)__doc__(ray.rllib.algorithms.dreamerv3.dreamerv3r   ray.rllib.utils.test_utilsr   parser
parse_argsargsenvironmentr   env_runnersr   num_learners	reportingtrainingconfig__name__r    r)   r)   a/home/ubuntu/.local/lib/python3.10/site-packages/ray/rllib/tuned_examples/dreamerv3/atari_100k.py<module>   sL    
!*