fix some hint bugs

Author: lixinqi

.pre-commit-config.yaml

@@ -11,17 +11,11 @@ repos:
     - id: ruff-check
       args: [--fix, --exit-non-zero-on-fix, --no-cache]
 
-- repo: https://github.com/PFCCLab/typos-pre-commit-mirror.git
-  rev: v1.39.2
-  hooks:
-    - id: typos
-      args: [--force-exclude]
-
 - repo: https://github.com/Lucas-C/pre-commit-hooks.git
   rev: v1.5.1
   hooks:
     - id: remove-crlf
     - id: remove-tabs
       name: Tabs remver (Python)
       files: (.*\.(py|bzl)|BUILD|.*\.BUILD|WORKSPACE)$
-      args: [--whitespaces-count, '4']
+      args: [--whitespaces-count, '4']

graph_net/torch/constraint_util.py

@@ -18,7 +18,6 @@ def __init__(self, config):
         if config is None:
             config = {}
 
-        graph_net_root = os.path.dirname(graph_net.__file__)
         decorator_config = {"use_dummy_inputs": True}
         self.predicator = RunModelPredicator(decorator_config)
 

graph_net/torch/dim_gen_passes/batch_call_method_view_pass.py

@@ -1,4 +1,3 @@
-import torch
 import torch.fx as fx
 from graph_net.torch.dim_gen_passes import DimensionGeneralizationPass
 import os

graph_net/torch/dim_gen_passes/naive_call_method_view_pass.py

@@ -1,4 +1,3 @@
-import torch
 import torch.fx as fx
 from graph_net.torch.dim_gen_passes import DimensionGeneralizationPass
 import os
@@ -22,7 +21,7 @@ def _node_need_rewrite(self, node) -> bool:
         if not (node.target == "view"):
             return False
         print(f"{self.dim=} {node.args[1:]=}")
-        if not (self.dim in node.args[1:]):
+        if self.dim not in node.args[1:]:
             return False
         return True
 

graph_net/torch/dim_gen_passes/tuple_arg_call_method_view_pass.py

@@ -1,4 +1,3 @@
-import torch
 import torch.fx as fx
 from graph_net.torch.dim_gen_passes import DimensionGeneralizationPass
 import os
@@ -25,7 +24,7 @@ def _node_need_rewrite(self, node) -> bool:
             return False
         if not (isinstance(node.args[1], tuple)):
             return False
-        if not (self.dim in node.args[1]):
+        if self.dim not in node.args[1]:
             return False
         return True
 

graph_net/torch/static_to_dynamic.py

@@ -1,20 +1,15 @@
 import traceback
 import logging
 import torch
-import torch.fx as fx
 from graph_net.torch.utils import get_dummy_named_tensors
 from torch.fx.passes.shape_prop import ShapeProp
 from graph_net.torch.utils import apply_templates
 from pathlib import Path
 import inspect
-from typing import Any
-from contextlib import contextmanager
-from torch.export import export
 from graph_net.torch.fx_graph_parse_util import parse_sole_graph_module
 from graph_net.torch.fx_graph_cache_util import (
     parse_immutable_model_path_into_sole_graph_module,
 )
-from graph_net.imp_util import load_module
 import os
 
 
@@ -51,12 +46,12 @@ def make_config(self, pass_names=()):
                 "naive_call_method_reshape_pass",
                 "naive_call_method_expand_pass",
                 "non_batch_call_method_expand_pass",
-                "non_batch_call_function_arange_pass",
+                "non_batch_call_function_arange_pass",  # typos: skip
                 "non_batch_call_function_getitem_slice_pass",
                 "non_batch_call_function_full_pass",
                 "non_batch_call_function_full_plus_one_pass",
                 "non_batch_call_function_zeros_pass",
-                "non_batch_call_function_arange_plus_one_pass",
+                "non_batch_call_function_arange_plus_one_pass",  # typos: skip
             )
         return {
             "pass_names": pass_names,
@@ -71,7 +66,7 @@ def need_rewrite(self, inputs):
             traced_module = self._create_fx_graph_module(inputs)
             logging.warning("after _create_fx_graph_module")
             ShapeProp(traced_module).propagate(*inputs)
-        except Exception as e:
+        except Exception:
             traceback.print_exc()
             return False
         return any(
@@ -112,4 +107,4 @@ def get_conditional_passes(self):
         ]
 
     def forward(self, *args, **kwargs):
-        print(f"Do nothing.")
+        print("Do nothing.")