diff options
Diffstat (limited to 'osdf/optimizers/pciopt/solver/optimizer.py')
| -rw-r--r-- | osdf/optimizers/pciopt/solver/optimizer.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/osdf/optimizers/pciopt/solver/optimizer.py b/osdf/optimizers/pciopt/solver/optimizer.py index 91e693c..5a1a5c2 100644 --- a/osdf/optimizers/pciopt/solver/optimizer.py +++ b/osdf/optimizers/pciopt/solver/optimizer.py @@ -64,7 +64,7 @@ def add_to_neighbor_list(network_cell_info, cell, neighbor_list): for nbr in cell.get('nbr_list', []): host_id = cell['id'] nbr_id = get_id(network_cell_info, nbr['cellId']) - if host_id != nbr_id: + if nbr_id and host_id != nbr_id: entry = sorted([host_id, nbr_id]) neighbor_list.add((entry[0], entry[1])) @@ -74,8 +74,9 @@ def get_second_level_neighbor(network_cell_info): for cell in network_cell_info['cell_list']: comb_list = build_second_level_list(network_cell_info, cell) for comb in comb_list: - s = sorted(comb) - second_neighbor_list.add((s[0], s[1])) + if comb[0] and comb[1]: + s = sorted(comb) + second_neighbor_list.add((s[0], s[1])) return sorted(second_neighbor_list) |