Add RenamedDataInputPredicator to support generate unittest for renamed subgraphs.

Author: Xreki

graph_net/torch/constraint_util.py

@@ -16,6 +16,14 @@ def __call__(self, model_path, input_var_name: str) -> bool:
         )
 
 
+class RenamedDataInputPredicator:
+    def __init__(self, config):
+        self.config = config
+
+    def __call__(self, model_path, input_var_name: str) -> bool:
+        return input_var_name.startswith("in_")
+
+
 class ModelRunnablePredicator:
     def __init__(self, config):
         if config is None:

graph_net/torch/sample_passes/agent_unittest_generator.py

@@ -143,7 +143,7 @@ def __init__(
 
     def generate(self):
         model_name = "".join(
-            word.capitalize() for word in re.split(r"[_-]", self.model_path.name)
+            word.capitalize() for word in re.split(r"[_.-]", self.model_path.name)
         )
         graph_module = load_class_from_file(
             self.model_path / "model.py", class_name="GraphModule"