Skip to content

Commit 9e58856

Browse files
committed
add __repr__ method for InsertableOrderedDict
1 parent 45392cc commit 9e58856

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

src/diffusers/modular_pipelines/modular_pipeline_utils.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,16 @@ def insert(self, key, value, index):
4141

4242
# Return self for method chaining
4343
return self
44+
45+
def __repr__(self):
46+
if not self:
47+
return "InsertableOrderedDict()"
48+
49+
items = []
50+
for i, (key, value) in enumerate(self.items()):
51+
items.append(f"{i}: ({repr(key)}, {repr(value)})")
52+
53+
return "InsertableOrderedDict([\n " + ",\n ".join(items) + "\n])"
4454

4555

4656
# YiYi TODO:

0 commit comments

Comments
 (0)