Skip to content

Commit 0daa98f

Browse files
committed
Fixed dict coercion with refType
1 parent 6bd4aaf commit 0daa98f

1 file changed

Lines changed: 11 additions & 4 deletions

File tree

runware/types.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1340,10 +1340,17 @@ def __post_init__(self):
13401340
if self.referenceImages is None:
13411341
self.referenceImages = self.references
13421342
if self.referenceImages is not None:
1343-
self.referenceImages = [
1344-
IInputReference(**item) if isinstance(item, dict) else item
1345-
for item in self.referenceImages
1346-
]
1343+
coerced_reference_images: List[Union[str, File, IInputReference]] = []
1344+
for item in self.referenceImages:
1345+
if isinstance(item, dict):
1346+
d = dict(item)
1347+
if "type" in d and "refType" not in d:
1348+
d["refType"] = d["type"]
1349+
d.pop("type", None)
1350+
coerced_reference_images.append(IInputReference(**d))
1351+
else:
1352+
coerced_reference_images.append(item)
1353+
self.referenceImages = coerced_reference_images
13471354
if self.fonts is not None:
13481355
self.fonts = [
13491356
IInputFont(**item) if isinstance(item, dict) else item

0 commit comments

Comments
 (0)