From 8177096ef687e4f7dae6a77d9538bc5973b72c61 Mon Sep 17 00:00:00 2001 From: Harry Mellor <19981378+hmellor@users.noreply.github.com> Date: Fri, 20 Mar 2026 12:51:20 +0100 Subject: [PATCH] Add missing dunder methods to `SizeDict` Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> --- src/transformers/image_utils.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/src/transformers/image_utils.py b/src/transformers/image_utils.py index 1691a485d3b4..0312bec9cea6 100644 --- a/src/transformers/image_utils.py +++ b/src/transformers/image_utils.py @@ -1001,3 +1001,17 @@ def __eq__(self, other): getattr(other, f.name) for f in fields(self) ) return NotImplemented + + def __or__(self, other) -> "SizeDict": + if isinstance(other, dict | SizeDict): + merged = dict(self) + merged.update(dict(other)) + return SizeDict(**merged) + return NotImplemented + + def __ror__(self, other) -> dict: + if isinstance(other, dict): + merged = dict(other) + merged.update(dict(self)) + return merged + return NotImplemented