revert some change in decompose_util

Author: JewelRoam

graph_net/torch/decompose_util.py

@@ -254,15 +254,7 @@ def get_args_node(arg):
             pass
         else:
             assert isinstance(
-                arg,
-                (
-                    int,
-                    bool,
-                    float,
-                    str,
-                    type(...),
-                    type(None),
-                ),
+                arg, (int, bool, float, str, type(...), type(None))
             ), f"{type(arg)=}"
 
     def get_args_node_and_self_node(node):