From 21dd0c3125c173f85593258737ee8108d7af85a4 Mon Sep 17 00:00:00 2001
From: Yakup Budanaz <budanaz.yakup@gmail.com>
Date: Tue, 3 Dec 2024 16:54:28 +0100
Subject: [PATCH] Name update fixes

---
 dace/sdfg/sdfg.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/dace/sdfg/sdfg.py b/dace/sdfg/sdfg.py
index bfa85dc6ed..488bbcae72 100644
--- a/dace/sdfg/sdfg.py
+++ b/dace/sdfg/sdfg.py
@@ -773,9 +773,11 @@ def replace_dict(self,
 
         # Update size descriptors
         # Return_size break things delete it from the arrays
+        # Changes symbol names might not related to arrays
         for arr_name, size_desc_name in size_desc_map.items():
-            arr = self.arrays[arr_name]
-            arr.size_desc_name = size_desc_name if size_desc_name != "__return_size" else None
+            arr = self.arrays[arr_name] if arr_name in self.arrays else None
+            if arr is not None:
+                arr.size_desc_name = size_desc_name if size_desc_name != "__return_size" else None
 
         # Replace inside data descriptors
         for array in self.arrays.values():