Linting fix

apache · Jan 21, 2022 · 1e9f642 · 1e9f642
1 parent 901f766
commit 1e9f642
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/python/tvm/relay/frontend/keras.py b/python/tvm/relay/frontend/keras.py
@@ -632,13 +632,11 @@ def _convert_pooling(
 
     if pool_type == "GlobalMaxPooling2D":
         return _convert_flatten(
-            _op.nn.global_max_pool2d(inexpr, **global_pool_params),
-            keras_layer, etab, data_layout
+            _op.nn.global_max_pool2d(inexpr, **global_pool_params), keras_layer, etab, data_layout
         )
     if pool_type == "GlobalAveragePooling2D":
         return _convert_flatten(
-            _op.nn.global_avg_pool2d(inexpr, **global_pool_params),
-            keras_layer, etab, data_layout
+            _op.nn.global_avg_pool2d(inexpr, **global_pool_params), keras_layer, etab, data_layout
         )
     pool_h, pool_w = keras_layer.pool_size
     stride_h, stride_w = keras_layer.strides