vertiport_autonomy.agents package

Subpackages

Submodules

vertiport_autonomy.agents.base module

Abstract base class for all agents.

class vertiport_autonomy.agents.base.BaseAgent(name: str)[source]

Bases: ABC

Abstract base class for vertiport autonomy agents.

__init__(name: str)[source]

Initialize the agent.

Parameters:

name – Human-readable name for the agent

abstractmethod act(observation: Dict[str, Any]) ndarray[source]

Select actions based on the current observation.

Parameters:

observation – Current environment observation

Returns:

Array of actions for each drone

abstractmethod reset() None[source]

Reset agent state for a new episode.

__str__() str[source]

String representation of the agent.

vertiport_autonomy.agents.heuristic module

Simplified Heuristic Baseline Agent

Provides a working baseline for comparison with DRL agents. Uses simple FCFS strategy with basic conflict avoidance.

class vertiport_autonomy.agents.heuristic.SimpleHeuristicAgent(name: str = 'SimpleHeuristic')[source]

Bases: BaseAgent

Simple heuristic agent that uses FCFS strategy.

__init__(name: str = 'SimpleHeuristic')[source]

Initialize the simple heuristic agent.

act(observation: Dict[str, Any]) ndarray[source]

Select actions based on simple heuristic strategy.

Parameters:

observation – Current environment observation

Returns:

Array of actions for each drone

reset() None[source]

Reset agent state for a new episode.

vertiport_autonomy.agents.heuristic.run_simple_heuristic(scenario_path: str, max_steps: int = 200)[source]

Run simple heuristic agent on given scenario

vertiport_autonomy.agents.heuristic.main()[source]

Test simple heuristic on all scenarios

Module contents

Agent implementations.

class vertiport_autonomy.agents.BaseAgent(name: str)[source]

Bases: ABC

Abstract base class for vertiport autonomy agents.

__init__(name: str)[source]

Initialize the agent.

Parameters:

name – Human-readable name for the agent

__str__() str[source]

String representation of the agent.

abstractmethod act(observation: Dict[str, Any]) ndarray[source]

Select actions based on the current observation.

Parameters:

observation – Current environment observation

Returns:

Array of actions for each drone

abstractmethod reset() None[source]

Reset agent state for a new episode.

class vertiport_autonomy.agents.SimpleHeuristicAgent(name: str = 'SimpleHeuristic')[source]

Bases: BaseAgent

Simple heuristic agent that uses FCFS strategy.

__init__(name: str = 'SimpleHeuristic')[source]

Initialize the simple heuristic agent.

act(observation: Dict[str, Any]) ndarray[source]

Select actions based on simple heuristic strategy.

Parameters:

observation – Current environment observation

Returns:

Array of actions for each drone

reset() None[source]

Reset agent state for a new episode.