Correct OrderChangeOffer model

The `add` and `remove` fields are lists.
duffelhq · Dec 16, 2021 · 0f24de5 · 0f24de5
1 parent 32e895b
commit 0f24de5
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/duffel_api/models/order_change_offer.py b/duffel_api/models/order_change_offer.py
@@ -44,8 +44,13 @@ class OrderChangeOfferSlices:
     """The slices to be added and/or removed"""
 
     def __init__(self, json):
-        setattr(self, "add", OrderChangeOfferSlicesAdd(json["add"]))
-        setattr(self, "remove", OrderChangeOfferSlicesRemove(json["remove"]))
+        add = list(map(lambda add: OrderChangeOfferSlicesAdd(add), json["add"]))
+        remove = list(
+            map(lambda remove: OrderChangeOfferSlicesRemove(remove), json["remove"])
+        )
+
+        setattr(self, "add", add)
+        setattr(self, "remove", remove)
 
 
 class OrderChangeOfferSlicesAdd: