diff --git a/tools/run_tests/xds_k8s_test_driver/framework/infrastructure/traffic_director.py b/tools/run_tests/xds_k8s_test_driver/framework/infrastructure/traffic_director.py index 6e8bbfed919..5f39ff0a129 100644 --- a/tools/run_tests/xds_k8s_test_driver/framework/infrastructure/traffic_director.py +++ b/tools/run_tests/xds_k8s_test_driver/framework/infrastructure/traffic_director.py @@ -556,12 +556,14 @@ class TrafficDirectorAppNetManager(TrafficDirectorManager): *, resource_prefix: str, resource_suffix: Optional[str] = None, - network: str = 'default'): + network: str = 'default', + compute_api_version: str = 'v1'): super().__init__(gcp_api_manager, project, resource_prefix=resource_prefix, resource_suffix=resource_suffix, - network=network) + network=network, + compute_api_version=compute_api_version) # API self.netsvc = _NetworkServicesV1Alpha1(gcp_api_manager, project) @@ -656,12 +658,14 @@ class TrafficDirectorSecureManager(TrafficDirectorManager): resource_prefix: str, resource_suffix: Optional[str] = None, network: str = 'default', + compute_api_version: str = 'v1', ): super().__init__(gcp_api_manager, project, resource_prefix=resource_prefix, resource_suffix=resource_suffix, - network=network) + network=network, + compute_api_version=compute_api_version) # API self.netsec = _NetworkSecurityV1Beta1(gcp_api_manager, project)