modify codestyle

Author: G2uge

graph_net/torch/backend/unstable_to_stable_backend.py

@@ -49,7 +49,11 @@ def replace_in_graph(graph_mod):
 
         # 处理主 gm 及所有嵌套的 GraphModule
         modules = [gm]
-        modules += [m for _, m in gm.named_modules() if isinstance(m, torch.fx.GraphModule) and m is not gm]
+        modules += [
+            m
+            for _, m in gm.named_modules()
+            if isinstance(m, torch.fx.GraphModule) and m is not gm
+        ]
         for m in modules:
             replace_in_graph(m)
 
@@ -94,11 +98,17 @@ def check_graph(graph_mod):
 
         # 检查主 gm 及所有嵌套的 GraphModule
         modules = [gm]
-        modules += [m for _, m in gm.named_modules() if isinstance(m, torch.fx.GraphModule) and m is not gm]
+        modules += [
+            m
+            for _, m in gm.named_modules()
+            if isinstance(m, torch.fx.GraphModule) and m is not gm
+        ]
         for m in modules:
             check_graph(m)
 
-        print(f"✅ Model passed: no occurrence of '{self.unstable_api}' found in graph nodes.")
+        print(
+            f"✅ Model passed: no occurrence of '{self.unstable_api}' found in graph nodes."
+        )
 
     def synchronize(self):
         # Synchronize CUDA operations if available