diff --git a/tables/automl/automl_tables_set_endpoint.py b/tables/automl/automl_tables_set_endpoint.py index fd865c9cac4d..d6ab898b4f5d 100644 --- a/tables/automl/automl_tables_set_endpoint.py +++ b/tables/automl/automl_tables_set_endpoint.py @@ -27,4 +27,7 @@ def create_client_with_endpoint(gcp_project_id): ) # [END automl_set_endpoint] + # do simple test to check client connectivity + print(client.list_datasets()) + return client diff --git a/tables/automl/endpoint_test.py b/tables/automl/endpoint_test.py index e00c24d9d692..b556a4c9827c 100644 --- a/tables/automl/endpoint_test.py +++ b/tables/automl/endpoint_test.py @@ -21,7 +21,6 @@ def test_client_creation(capsys): - client = automl_tables_set_endpoint.create_client_with_endpoint(PROJECT) - print(client.list_datasets()) + automl_tables_set_endpoint.create_client_with_endpoint(PROJECT) out, _ = capsys.readouterr() assert "GRPCIterator" in out