From 3ed4fd64df68e9ffd7a618c45fc483b2c49f24cd Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Tue, 8 Oct 2019 14:27:55 +0800 Subject: [PATCH] fix prune interface (#3472) --- tools/infer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer.py b/tools/infer.py index d33d1e3eb..64049e3fc 100644 --- a/tools/infer.py +++ b/tools/infer.py @@ -106,7 +106,7 @@ def prune_feed_vars(feeded_var_names, target_vars, prog): """ exist_var_names = [] prog = prog.clone() - prog = prog._prune(feeded_var_names, targets=target_vars) + prog = prog._prune(targets=target_vars) global_block = prog.global_block() for name in feeded_var_names: try: -- GitLab