From 6a7e7ce1b1460389646985ec69d41bfd8e36f481 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Sat, 9 May 2020 16:58:33 +0800 Subject: [PATCH] fix(mge/cgtools): make get_dep_vars accept symbolvar input correctly GitOrigin-RevId: 8bf00a3006599e2366204f4743855569f363ac41 --- python_module/megengine/_internal/comp_graph_tools.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python_module/megengine/_internal/comp_graph_tools.py b/python_module/megengine/_internal/comp_graph_tools.py index bbbd3ef2..19223d07 100644 --- a/python_module/megengine/_internal/comp_graph_tools.py +++ b/python_module/megengine/_internal/comp_graph_tools.py @@ -25,7 +25,7 @@ def get_dep_vars(var, var_type=None): outputs = [] memo = set() - if not isinstance(var, collections.Iterable): + if isinstance(var, _mgb.SymbolVar): var = [var] if isinstance(var_type, str):