o
    ci	                     @   s  d Z ddlmZ ddlmZ dZe ZejZej	Z
dd Zede edjddjedkr2dneer7dndd	jd
ep?d ddjepGdddddjdddepSd dde gdde gdde ggdde
 gdde
 gdde
 ggdde
 gdde
 gdde
 ggd 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)tunec                 C   sD   dd l }dd l}ddlm} ddlm} |||jddddddS )	Nr   )	resize_v1)NormalizedImageEnvzFlappyBird-rgb-v0F)audio_on@   )x_sizey_size)flappy_bird_gymnasium	gymnasiumsupersuit.generic_wrappersr   /ray.rllib.algorithms.dreamerv3.utils.env_runnerr   make)ctxr
   gymr   r    r   b/home/ubuntu/.local/lib/python3.10/site-packages/ray/rllib/tuned_examples/dreamerv3/flappy_bird.py_env_creator   s   r   zflappy-bird   )num_cpus_for_main_process)num_learnersnum_gpus_per_learner   T)num_envs_per_env_runnerremote_worker_envsF)"metrics_num_episodes_for_smoothingreport_images_and_videosreport_dream_data"report_individual_batch_item_statsMr      g?i@  i'     )
model_sizetraining_ratiobatch_size_Bworld_model_lr	critic_lractor_lrN)__doc__(ray.rllib.algorithms.dreamerv3.dreamerv3r   rayr   num_gpusconfigr%   wr&   cr   register_envenvironment	resourceslearnersenv_runners	reportingtrainingr   r   r   r   <module>   sF    


   
