fix and enable flake8 E721

GreatV · May 25, 2024 · add3f6d · add3f6d
1 parent 230da30
commit add3f6d
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 12 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -43,7 +43,7 @@ repos:
     -   id: flake8
         args:
             - --count
-            - --select=E9,F63,F7,F82
+            - --select=E9,F63,F7,F82,E721
             - --show-source
             - --statistics
         exclude: ^benchmark/|^test_tipc/

diff --git a/ppocr/losses/distillation_loss.py b/ppocr/losses/distillation_loss.py
@@ -78,9 +78,9 @@ def _check_model_name_pairs(self, model_name_pairs):
     def _check_maps_name(self, maps_name):
         if maps_name is None:
             return None
-        elif type(maps_name) == str:
+        elif isinstance(maps_name, str):
             return [maps_name]
-        elif type(maps_name) == list:
+        elif isinstance(maps_name, list):
             return [maps_name]
         else:
             return None
@@ -174,9 +174,9 @@ def _check_model_name_pairs(self, model_name_pairs):
     def _check_maps_name(self, maps_name):
         if maps_name is None:
             return None
-        elif type(maps_name) == str:
+        elif isinstance(maps_name, str):
             return [maps_name]
-        elif type(maps_name) == list:
+        elif isinstance(maps_name, list):
             return [maps_name]
         else:
             return None
@@ -282,9 +282,9 @@ def _check_model_name_pairs(self, model_name_pairs):
     def _check_maps_name(self, maps_name):
         if maps_name is None:
             return None
-        elif type(maps_name) == str:
+        elif isinstance(maps_name, str):
             return [maps_name]
-        elif type(maps_name) == list:
+        elif isinstance(maps_name, list):
             return [maps_name]
         else:
             return None
@@ -428,9 +428,9 @@ def _check_model_name_pairs(self, model_name_pairs):
     def _check_maps_name(self, maps_name):
         if maps_name is None:
             return None
-        elif type(maps_name) == str:
+        elif isinstance(maps_name, str):
             return [maps_name]
-        elif type(maps_name) == list:
+        elif isinstance(maps_name, list):
             return [maps_name]
         else:
             return None
@@ -536,9 +536,9 @@ def _check_model_name_pairs(self, model_name_pairs):
     def _check_maps_name(self, maps_name):
         if maps_name is None:
             return None
-        elif type(maps_name) == str:
+        elif isinstance(maps_name, str):
             return [maps_name]
-        elif type(maps_name) == list:
+        elif isinstance(maps_name, list):
             return [maps_name]
         else:
             return None

diff --git a/tools/program.py b/tools/program.py
@@ -209,7 +209,7 @@ def train(
     if "global_step" in pre_best_model_dict:
         global_step = pre_best_model_dict["global_step"]
     start_eval_step = 0
-    if type(eval_batch_step) == list and len(eval_batch_step) >= 2:
+    if isinstance(eval_batch_step, list) and len(eval_batch_step) >= 2:
         start_eval_step = eval_batch_step[0] if not eval_batch_epoch else 0
         eval_batch_step = (
             eval_batch_step[1]