From 9e58856b7a80cf995307be62ef6dca839ffa11c6 Mon Sep 17 00:00:00 2001 From: yiyixuxu Date: Sat, 21 Jun 2025 04:24:44 +0200 Subject: [PATCH] add __repr__ method for InsertableOrderedDict --- .../modular_pipelines/modular_pipeline_utils.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/src/diffusers/modular_pipelines/modular_pipeline_utils.py b/src/diffusers/modular_pipelines/modular_pipeline_utils.py index ced059551f..868c091060 100644 --- a/src/diffusers/modular_pipelines/modular_pipeline_utils.py +++ b/src/diffusers/modular_pipelines/modular_pipeline_utils.py @@ -41,6 +41,16 @@ class InsertableOrderedDict(OrderedDict): # Return self for method chaining return self + + def __repr__(self): + if not self: + return "InsertableOrderedDict()" + + items = [] + for i, (key, value) in enumerate(self.items()): + items.append(f"{i}: ({repr(key)}, {repr(value)})") + + return "InsertableOrderedDict([\n " + ",\n ".join(items) + "\n])" # YiYi TODO: