From 639cf04e688f4e498d7140c38d204a75f4bc6575 Mon Sep 17 00:00:00 2001 From: weguo-NV <154216071+weiguo-nvidia@users.noreply.github.com> Date: Tue, 8 Jul 2025 02:36:34 +0800 Subject: [PATCH] Fix github issue #16529 (#18117) --- tests/sub_port_interfaces/conftest.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/tests/sub_port_interfaces/conftest.py b/tests/sub_port_interfaces/conftest.py index ae05e45e7e7..083d5f86419 100644 --- a/tests/sub_port_interfaces/conftest.py +++ b/tests/sub_port_interfaces/conftest.py @@ -523,7 +523,15 @@ def apply_balancing_config(duthost, ptfhost, ptfadapter, define_sub_ports_config all_up_ports = set() for port in list(mg_facts['minigraph_ports'].keys()): all_up_ports.add("eth" + str(mg_facts['minigraph_ptf_indices'][port])) - src_ports = tuple(all_up_ports.difference(ptf_ports)) + + # Handle both list and dict ptf_ports + ports_to_exclude = set() + if isinstance(ptf_ports, dict): + ports_to_exclude = set(ptf_ports.values()) + else: + ports_to_exclude = set(ptf_ports) + + src_ports = tuple(all_up_ports.difference(ports_to_exclude)) network = '1.1.1.0/24' network = ipaddress.ip_network(network)