**********************
Task 2: Rearrange Dice
**********************


.. image:: images/trifingerpro_with_dice.jpg
   :alt: TriFinger robot with multiple dice in the arena


.. automodule:: trifinger_simulation.tasks.rearrange_dice

   .. autodata:: EPISODE_LENGTH
   .. autodata:: NUM_DICE
   .. autofunction:: sample_goal
   .. autofunction:: evaluate_state
   .. autofunction:: validate_goal
   .. autofunction:: json_goal_from_config
   .. autofunction:: seed()
   .. autofunction:: visualize_2d

       Example on how the visualisation look like:

       .. image:: /images/rearrange_dice_visualize_2d.png
          :width: 50%
          :align: center

   .. autofunction:: generate_goal_mask

       Example on how the generated goal masks look like:

       .. image:: /images/rearrange_dice_goal_mask.png