Skip to content

Commit 0609b45

Browse files
committed
Add __repr__ for CustomTransferMap
1 parent d48bbe9 commit 0609b45

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

cheetah/accelerator.py

+7
Original file line numberDiff line numberDiff line change
@@ -234,6 +234,13 @@ def broadcast(self, shape: Size) -> Element:
234234
def is_skippable(self) -> bool:
235235
return True
236236

237+
def __repr__(self):
238+
return (
239+
f"{self.__class__.__name__}(transfer_map={repr(self._transfer_map)}, "
240+
+ f"length={repr(self.length)}, "
241+
+ f"name={repr(self.name)})"
242+
)
243+
237244
def defining_features(self) -> list[str]:
238245
return super().defining_features + ["transfer_map"]
239246

0 commit comments

Comments
 (0)