From 802b44c1f209c806bea0e2ce9cf261ed9ef4c739 Mon Sep 17 00:00:00 2001
From: Raffael Meyer <14891507+barredterra@users.noreply.github.com>
Date: Wed, 22 Jan 2025 11:52:28 +0100
Subject: [PATCH] fix: supported profile for TradeDelivery members (#56)

---
 drafthorse/models/delivery.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/drafthorse/models/delivery.py b/drafthorse/models/delivery.py
index c51ab96..b584cc8 100644
--- a/drafthorse/models/delivery.py
+++ b/drafthorse/models/delivery.py
@@ -1,4 +1,4 @@
-from . import BASIC, EXTENDED, NS_RAM
+from . import BASIC, COMFORT, EXTENDED, NS_RAM
 from .elements import Element
 from .fields import DateTimeField, Field, StringField
 from .party import (
@@ -49,10 +49,11 @@ class TradeDelivery(Element):
         SupplyChainConsignment,
         default=False,
         required=False,
+        profile=EXTENDED,
         _d="Detailinformationen zur Konsignation oder Sendung",
     )
     ship_to: ShipToTradeParty = Field(
-        ShipToTradeParty, required=False, profile=EXTENDED
+        ShipToTradeParty, required=False, profile=COMFORT
     )
     ultimate_ship_to: UltimateShipToTradeParty = Field(
         UltimateShipToTradeParty, required=False, profile=EXTENDED
@@ -62,7 +63,7 @@ class TradeDelivery(Element):
     )
     event: SupplyChainEvent = Field(SupplyChainEvent, required=False, profile=BASIC)
     despatch_advice: DespatchAdviceReferencedDocument = Field(
-        DespatchAdviceReferencedDocument, required=False, profile=EXTENDED
+        DespatchAdviceReferencedDocument, required=False, profile=COMFORT
     )
     delivery_note: DeliveryNoteReferencedDocument = Field(
         DeliveryNoteReferencedDocument, required=False, profile=EXTENDED
-- 
2.47.3