[F] Fix pylint error

DocsaidLab · Dec 21, 2024 · ea95495 · ea95495
1 parent c3cc3f6
commit ea95495
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/.pylintrc b/.github/workflows/.pylintrc
@@ -89,7 +89,7 @@ enable=
     E0601,E0602,E0603,E0604,E0611,
     E0632,E0633,E0701,E0702,E0703,
     E0704,E0710,E0711,E0712,E1003,
-    #E1102,
+    ; #E1102,
     E1111,E1120,E1121,E1123,
     E1124,E1125,E1126,E1127,E1128,
     E1129,E1130,E1131,E1132,E1133,

diff --git a/chameleon/tools/calflops/calculate_pipline.py b/chameleon/tools/calflops/calculate_pipline.py
@@ -254,7 +254,7 @@ def flops_repr(module):
             return ", ".join(items)
 
         def add_extra_repr(module):
-            flops_extra_repr = flops_repr.__get__(module)
+            flops_extra_repr = flops_repr(module)
             if module.extra_repr != flops_extra_repr:
                 module.original_extra_repr = module.extra_repr
                 module.extra_repr = flops_extra_repr
@@ -349,7 +349,7 @@ def flops_repr(module):
             return ", ".join(items)
 
         def add_extra_repr(module):
-            flops_extra_repr = flops_repr.__get__(module)
+            flops_extra_repr = flops_repr(module)
             if module.extra_repr != flops_extra_repr:
                 module.original_extra_repr = module.extra_repr
                 module.extra_repr = flops_extra_repr

diff --git a/chameleon/tools/calflops/flops_counter.py b/chameleon/tools/calflops/flops_counter.py
@@ -171,9 +171,9 @@ def calculate_flops(model,
     params = calculate_flops_pipline.get_total_params()
 
     if print_results:
-        return_print = calculate_flops_pipline.print_model_pipline(units=output_unit,
-                                                                   precision=output_precision,
-                                                                   print_detailed=print_detailed)
+        calculate_flops_pipline.print_model_pipline(units=output_unit,
+                                                    precision=output_precision,
+                                                    print_detailed=print_detailed)
 
     calculate_flops_pipline.end_flops_calculate()