Skip to content

Commit 978fb48

Browse files
committed
Change AgentUnittestGenerator to a common implementation for both torch and paddle.
1 parent 1c34917 commit 978fb48

File tree

4 files changed

+152
-253
lines changed

4 files changed

+152
-253
lines changed
Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
import logging
2+
3+
logger = logging.getLogger(__name__)
4+
5+
6+
class NaiveDataInputPredicator:
7+
def __init__(self, config):
8+
self.config = config
9+
10+
def __call__(self, model_path, input_var_name: str) -> bool:
11+
return not input_var_name.startswith("parameter_")

graph_net/paddle/sample_passes/agent_unittest_generator.py

Lines changed: 0 additions & 233 deletions
This file was deleted.

0 commit comments

Comments
 (0)