diff --git a/sdks/python/apache_beam/transforms/enrichment_handlers/cloudsql_it_test.py b/sdks/python/apache_beam/transforms/enrichment_handlers/cloudsql_it_test.py index 15ab0ec0a3a1..b953b67841ac 100644 --- a/sdks/python/apache_beam/transforms/enrichment_handlers/cloudsql_it_test.py +++ b/sdks/python/apache_beam/transforms/enrichment_handlers/cloudsql_it_test.py @@ -209,6 +209,7 @@ def create_table( class BaseTestSQLEnrichment(unittest.TestCase): + _cache_client_retries = 3 _table_data = [ { "id": 1, "name": "A", 'quantity': 2, 'distribution_center_id': 3 @@ -259,8 +260,6 @@ def setUpClass(cls): table_data=cls._table_data, metadata=cls._metadata) - cls._cache_client_retries = 3 - @classmethod def get_columns(cls): """Returns fresh column objects each time it's called."""