Callback API

LoS Evaluation Callback

UrbanMARL Callback Module for Line-of-Sight Evaluation.

Provides BenchMARL experiment callbacks for tracking LoS and network metrics during training.

class urbanmarl.callback.evaluate_los_per_urban.EvaluateLoS[source]

Bases: Callback

BenchMARL Callback to evaluate LoS ratios per urban environment configuration.

Interprets training and evaluation batches to record LoS statistics.

on_train_step(batch, group)[source]

Invoked on each training step.

Parameters:

  • batch (TensorDictBase) – Tensordict containing training batch data.

  • group (str) – Agent group identifier name.

Returns:

Tensordict with optional computed metrics.

Return type:

TensorDictBase

on_train_end(training_td, group)[source]

Invoked at the conclusion of experiment training.

Parameters:

  • training_td (TensorDictBase) – Tensordict containing training history metrics.

  • group (str) – Agent group identifier name.

Return type:

None