From 86759cc47a90ccffd2e43cb36bdc04b523dfcc68 Mon Sep 17 00:00:00 2001
From: Lionel GUEZ <guez@lmd.ens.fr>
Date: Tue, 12 Sep 2023 19:09:01 +0200
Subject: [PATCH] Manage missing trajectories
This can happen if the input graph file contains only an extracted
component.
---
Trajectories/draw_segments.py | 10 +++++++---
1 file changed, 7 insertions(+), 3 deletions(-)
diff --git a/Trajectories/draw_segments.py b/Trajectories/draw_segments.py
index 6971b3a4..fa4758c8 100755
--- a/Trajectories/draw_segments.py
+++ b/Trajectories/draw_segments.py
@@ -49,9 +49,13 @@ if args.t:
color = next(colors)
for segment in segment_list:
- n = g_out.get_node(segment)
- n.attr["label"] = f"{segment} in {i}"
- n.attr["color"] = color
+ try:
+ n = g_out.get_node(segment)
+ except KeyError:
+ break
+ else:
+ n.attr["label"] = f"{segment} in {i}"
+ n.attr["color"] = color
g_out.write(args.output_file)
--
GitLab