diff --git a/components/gcp/container/component_sdk/python/kfp_component/google/dataproc/_client.py b/components/gcp/container/component_sdk/python/kfp_component/google/dataproc/_client.py index ebde7da557..4dab4052db 100644 --- a/components/gcp/container/component_sdk/python/kfp_component/google/dataproc/_client.py +++ b/components/gcp/container/component_sdk/python/kfp_component/google/dataproc/_client.py @@ -22,7 +22,7 @@ class DataprocClient: """ Internal client for calling Dataproc APIs. """ def __init__(self): - self._dataproc = discovery.build('dataproc', 'v1') + self._dataproc = discovery.build('dataproc', 'v1', cache_discovery=False) def create_cluster(self, project_id, region, cluster, request_id): """Creates a new dataproc cluster. @@ -115,4 +115,4 @@ class DataprocClient: self._dataproc.projects().regions().operations().cancel( name = operation_name - ).execute() \ No newline at end of file + ).execute() diff --git a/components/gcp/container/component_sdk/python/kfp_component/google/ml_engine/_client.py b/components/gcp/container/component_sdk/python/kfp_component/google/ml_engine/_client.py index 65a340191f..aff58d9437 100644 --- a/components/gcp/container/component_sdk/python/kfp_component/google/ml_engine/_client.py +++ b/components/gcp/container/component_sdk/python/kfp_component/google/ml_engine/_client.py @@ -23,7 +23,7 @@ class MLEngineClient: """ Client for calling MLEngine APIs. """ def __init__(self): - self._ml_client = discovery.build('ml', 'v1') + self._ml_client = discovery.build('ml', 'v1', cache_discovery=False) def create_job(self, project_id, job): """Create a new job. @@ -186,4 +186,4 @@ class MLEngineClient: """ self._ml_client.projects().operations().cancel( name = operation_name - ).execute() \ No newline at end of file + ).execute()