GraphTrainableFeaturesCallback plot produces unexpected artifacts for certain edge cases

[JPXKQX]

What happened?

The GraphTrainableFeaturesCallback plot for the edges produces strange artefacts.

The issue is when plotting an edge from (-0.05, y) to (0.05, y). The coordinates are expressed in the [0, 2pi] domain, so the graph creates a line between (0.05, y) and (2pi-0.05, y)

For example, when plotting an edge from (−0.05, y) to (0.05, y), the plot creates an artefact—a line drawn from (0.05,y) to (2π−0.05,y). This unexpected line appears due to the coordinate wrap-around, causing unintended visual connections across the map.

Expected behaviour

Plot two lines instead, one line from (0, y) to (0.05, y) and another from (2pi-0.05, y) to (2pi, y).

What are the steps to reproduce the bug?

Train a global weather model with GraphTrainableFeaturesCallback included.

Version

develop

Platform (OS and architecture)

x86_64 GNU/Linux

Relevant log output

No response

Accompanying data

No response

Organisation

ECMWF