refactoring: 'if' syntax deduplication

This commit is contained in:
marload 2020-04-07 14:55:53 +09:00
parent aae07a2470
commit 15f2821f87

View File

@ -363,10 +363,8 @@ def _get_grouped_variables(vars_to_warm_start):
# out the list.
grouped_variables = {}
for v in list_of_vars:
if not isinstance(v, list):
var_name = _infer_var_name([v])
else:
var_name = _infer_var_name(v)
v = [v] if not isinstance(v, list) else v
var_name = _infer_var_name(v)
grouped_variables.setdefault(var_name, []).append(v)
return grouped_variables