Skip to content

Commit 62c832d

Browse files
committed
make 10/10 for black formatting lol
more appeasing the formatting gods wow
1 parent f672f42 commit 62c832d

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

python/tvm/topi/testing/poolnd_python.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,11 @@
1414
# KIND, either express or implied. See the License for the
1515
# specific language governing permissions and limitations
1616
# under the License.
17-
17+
# pylint: disable=invalid-name, unused-argument, unused-variable
18+
"""Ground truth max and average pooling operators in python."""
1819
import itertools
1920
import math
20-
from typing import *
21+
from typing import List, Tuple
2122

2223
import numpy as np
2324
import tvm
@@ -61,6 +62,7 @@ def pad_tensor(
6162
padding_after: List[int],
6263
dtype: str,
6364
) -> np.array:
65+
"""Pad the spatial dimensions of the given array."""
6466
orig_shape = list(np_arr.shape)
6567
padded_shape = list(np_arr.shape)
6668
n = len(orig_shape)
@@ -110,7 +112,7 @@ def poolnd_python(
110112
out_shape.append(int(math.floor(val) + 1))
111113
out_shape = tuple(out_shape)
112114

113-
# Create a padded array, and a boolean mask showing which values are padded values and which are not
115+
# Create a padded array, and a boolean mask showing which values are padded values
114116
pad_value = 0
115117
if pool_type == "max" and not count_include_pad:
116118
pad_value = tvm.te.min_value(dtype).value

0 commit comments

Comments
 (0)