Skip to content

Commit

Permalink
auto format by CI
Browse files Browse the repository at this point in the history
  • Loading branch information
oneflow-ci-bot committed May 31, 2024
1 parent 4dc75dc commit f0e5e24
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/oneflow/nn/modules/pooling.py
Original file line number Diff line number Diff line change
Expand Up @@ -1116,8 +1116,8 @@ def _unpool_output_size_check(
assert (
indices.dtype == flow.int64
), f"elements in indices should be type int64 but got: {indices.dtype}"
assert (
(len(x.size()) == (expected_out_size + 1)) or (len(x.size()) == expected_out_size + 2)
assert (len(x.size()) == (expected_out_size + 1)) or (
len(x.size()) == expected_out_size + 2
), f"Input to max_unpooling1d should be a {expected_out_size + 1}d or {expected_out_size + 2}d Tensor, but got {len(x.size())} dimensions"
assert (
x.size() == indices.size()
Expand Down

0 comments on commit f0e5e24

Please sign in to comment.