From d9191c9ceb101a1bbd7be55eae91308128a39f06 Mon Sep 17 00:00:00 2001
From: ewuerger <ernst.wuerger@gmail.com>
Date: Fri, 13 Dec 2024 18:32:25 +0100
Subject: [PATCH] fix: Fix captions for custom diagrams

---
 capella2polarion/converters/element_converter.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/capella2polarion/converters/element_converter.py b/capella2polarion/converters/element_converter.py
index 82c11d5..ba2afa9 100644
--- a/capella2polarion/converters/element_converter.py
+++ b/capella2polarion/converters/element_converter.py
@@ -558,6 +558,11 @@ def _add_custom_diagrams(
                 650,
                 "additional-attributes-diagram",
                 cd_config.render_params,
+                (
+                    ("Figure", f"Context Diagram of {diagram.target.name}")
+                    if self.generate_figure_captions
+                    else None
+                ),
             )
             if attachment:
                 self._add_attachment(converter_data.work_item, attachment)