[Feature Enhancement] Opimize the unittest generator in llm format for torch. (#452)

* Support to generate unittest for llm.

* Add RenamedDataInputPredicator to support generate unittest for renamed subgraphs.

Author: Xreki

graph_net/sample_pass/sample_pass.py

@@ -55,7 +55,8 @@ def _make_config_by_config_declare(self, config):
         sig = inspect.signature(self.declare_config)
         mut_config = copy.deepcopy(config)
         for name, param in sig.parameters.items():
-            self._complete_default(name, param, mut_config)
+            if name not in mut_config:
+                self._complete_default(name, param, mut_config)
             class_name = type(self).__name__
             assert name in mut_config, f"{name=} {class_name=}"
 

…et/test/test_agent_unittest_generator.sh …et/test/agent_unittest_generator_test.shgraph_net/test/test_agent_unittest_generator.sh renamed to graph_net/test/agent_unittest_generator_test.sh graph_net/test/test_agent_unittest_generator.sh renamed to graph_net/test/agent_unittest_generator_test.sh

@@ -1,23 +1,22 @@
 #!/usr/bin/env bash
-# set -euo pipefail
-
-# Smoke tests for AgentUnittestGenerator using model_path_handler + sample pass.
 
 ROOT_DIR="$(cd "$(dirname "$0")/../.." && pwd)"
 GRAPH_NET_ROOT=$(python -c "import graph_net, os; print(os.path.dirname(os.path.dirname(graph_net.__file__)))")
-HANDLER_PATH="$GRAPH_NET_ROOT/graph_net/torch/sample_passes/agent_unittest_generator.py"
+
 MODEL_PATH_PREFIX="$ROOT_DIR"
-OUTPUT_DIR="$ROOT_DIR"
+OUTPUT_DIR="/tmp/agent_unittests"
 
 HANDLER_CONFIG=$(base64 -w 0 <<EOF
 {
-    "handler_path": "$HANDLER_PATH",
+    "handler_path": "$GRAPH_NET_ROOT/graph_net/torch/sample_passes/agent_unittest_generator.py",
     "handler_class_name": "AgentUnittestGeneratorPass",
     "handler_config": {
         "model_path_prefix": "$MODEL_PATH_PREFIX",
         "output_dir": "$OUTPUT_DIR",
-        "force_device": "auto",
-        "use_dummy_inputs": false
+        "device": "auto",
+        "generate_main": true,
+        "data_input_predicator_filepath": "$GRAPH_NET_ROOT/graph_net/torch/constraint_util.py",                                                                                     
+        "data_input_predicator_class_name": "NaiveDataInputPredicator"
     }
 }
 EOF

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: