From 604b37c3c6d8d65990a93074038b2716d5a026c1 Mon Sep 17 00:00:00 2001 From: Sangram Date: Mon, 29 Jul 2019 01:19:44 +0530 Subject: [PATCH] Fix create table deadline exceed error --- bigtable/docs/snippets.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/bigtable/docs/snippets.py b/bigtable/docs/snippets.py index 7f4071b41fd8..a2b4898b2121 100644 --- a/bigtable/docs/snippets.py +++ b/bigtable/docs/snippets.py @@ -403,7 +403,18 @@ def test_bigtable_create_table(): table = instance.table("table_my") # Define the GC policy to retain only the most recent 2 versions. max_versions_rule = column_family.MaxVersionsGCRule(2) - table.create(column_families={"cf1": max_versions_rule}) + + RETRIES = 4 + # Retry if deadline exceed error. + for retry in range(RETRIES): + try: + table.create(column_families={"cf1": max_versions_rule}) + break + except Exception as e: + if retry == RETRIES - 1: + raise e + else: + time.sleep(2 ** (retry - 1)) # [END bigtable_create_table] try: