def test_return_dict() -> None:
input_size = [torch.Size([1, 28, 28]), [12]]
metrics = summary(
ReturnDict(), input_size=input_size, col_width=65, batch_dim=0
)
assert metrics.input_size == [(1, 28, 28), [12]]
def test_return_dict(self):
input_size = [torch.Size([1, 28, 28]), [12]]
metrics = summary(ReturnDict(), input_size, col_width=65)
assert metrics.input_size == [(1, 28, 28), [12]]