LLM Version

[gioelemo]

This PR includes the following:

• Use custom starting position and starting angle for the Agent
• Slightly adapted visualization script to include more than one Agent (passed as additional information).
• Minor changes for LLM version (see PR in terra-baselines)

The executables

DATASET_PATH=/path/to/dataset DATASET_SIZE=<num_maps_per_type> python -m terra.viz.main_manual

and

DATASET_PATH=/path/to/dataset DATASET_SIZE=<num_maps_per_type> python -m terra.viz.play

should work as before

[Copilot]

Pull Request Overview

This PR adds support for custom agent positioning and angle initialization, along with enhanced visualization capabilities for multiple agents. The changes enable users to specify custom starting positions and angles for agents instead of relying solely on random or default positioning.

• Custom agent positioning and angle support through new optional parameters
• Enhanced visualization to handle multiple agents with partition visualization
• Configuration updates to simplify curriculum levels for LLM integration

Reviewed Changes

Copilot reviewed 6 out of 7 changed files in this pull request and generated 7 comments.

Show a summary per file

File	Description
terra/viz/main_manual.py	Adds custom position and angle parameters to environment reset
terra/viz/game/game.py	Implements multi-agent visualization with partition support and agent configuration handling
terra/state.py	Adds custom position and angle parameters to State creation methods
terra/env.py	Updates environment reset to support custom positioning parameters
terra/config.py	Simplifies curriculum configuration to focus on foundations maps
terra/agent.py	Implements comprehensive custom positioning logic with validation

---

_{Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.}

[Copilot]

The sentinel value -1 for unset angle should be defined as a constant to improve code clarity.

Suggested change

	temp_angle = jnp.full((1,), custom_angle, dtype=IntMap) if custom_angle is not None else jnp.full((1,), -1, dtype=IntMap)
	temp_angle = jnp.full((1,), custom_angle, dtype=IntMap) if custom_angle is not None else jnp.full((1,), UNSET_ANGLE, dtype=IntMap)

[Copilot]

The function docstring should document the expected types and ranges for the parameters, particularly the Array types and what constitutes valid position/angle values.