Fix linters

Author: sovrasov

.isort.cfg

@@ -1,5 +1,5 @@
 [isort]
-line_length = 79
+line_length = 89
 multi_line_output = 0
 known_standard_library = setuptools
 known_first_party = ptflops

ptflops/aten_engine.py

@@ -54,7 +54,7 @@ def f(*args):
 
     def exit_module(self, name):
         def f(*args):
-            assert(self.parents[-1] == name)
+            assert (self.parents[-1] == name)
             self.parents.pop()
         return f
 
@@ -138,7 +138,7 @@ def get_flops_aten(model, input_res,
 
     except Exception as e:
         print("Flops estimation was not finished successfully because of"
-              f" the following exception:\n{type(e)} : {e}")
+              f" the following exception: \n{type(e)}: {e}")
         traceback.print_exc()
 
         return None, None

ptflops/pytorch_engine.py

@@ -15,8 +15,8 @@
 import torch.nn as nn
 import torch.nn.functional as F
 
-from .pytorch_ops import (CUSTOM_MODULES_MAPPING, FUNCTIONAL_MAPPING,
-                          MODULES_MAPPING, TENSOR_OPS_MAPPING)
+from .pytorch_ops import (CUSTOM_MODULES_MAPPING, FUNCTIONAL_MAPPING, MODULES_MAPPING,
+                          TENSOR_OPS_MAPPING)
 from .utils import flops_to_string, params_to_string
 
 
@@ -72,7 +72,7 @@ def reset_environment():
 
     except Exception as e:
         print("Flops estimation was not finished successfully because of"
-              f" the following exception:\n{type(e)} : {e}")
+              f" the following exception: \n{type(e)}: {e}")
         traceback.print_exc()
         reset_environment()
 

ptflops/pytorch_ops.py

@@ -57,7 +57,8 @@ def bn_flops_counter_hook(module, input, output):
     module.__flops__ += int(batch_flops)
 
 
-def conv_flops_counter_hook(conv_module, input, output, extra_per_position_flops=0, transpose=False):
+def conv_flops_counter_hook(conv_module, input, output, extra_per_position_flops=0,
+                            transpose=False):
     # Can have multiple inputs, getting the first one
     input = input[0]
 

pyproject.toml

@@ -26,7 +26,7 @@ classifiers = [
 
 [project.optional-dependencies]
 dev = [
-  "flake8==3.8.1",
+  "flake8==5.0.1",
   "flake8-import-order==0.18.1",
   "isort==4.3.21",
   "torchvision>=0.5.0",