From cf2f2550efc284ff504608803a24034115871d17 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=C3=A9on=20Kuchenbecker?=
 <leon.kuchenbecker@uni-tuebingen.de>
Date: Tue, 14 May 2024 13:32:37 +0000
Subject: [PATCH] Apply suggestions from code review

Co-authored-by: Christoph Zwerschke <c.zwerschke@dkfz-heidelberg.de>
---
 .../builtin_transformations/infer_relations/model_transform.py  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/metldata/builtin_transformations/infer_relations/model_transform.py b/src/metldata/builtin_transformations/infer_relations/model_transform.py
index 202145e..41c9efa 100644
--- a/src/metldata/builtin_transformations/infer_relations/model_transform.py
+++ b/src/metldata/builtin_transformations/infer_relations/model_transform.py
@@ -160,5 +160,5 @@ def add_inferred_relations(
         )
 
     model_dict = model.model_dump()
-    model_dict.update({"classes": {**model.classes, **updated_class_defs}})
+    model_dict["classes"].update(updated_class_defs)
     return SchemaPack.model_validate(model_dict)