added line to make sure tables are removed

googleapis · Dec 14, 2023 · 6b7404a · 6b7404a
1 parent c206b77
commit 6b7404a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/samples/hello_happybase/main.py b/samples/hello_happybase/main.py
@@ -97,7 +97,12 @@ def main(project_id, instance_id, table_name):
 
     finally:
         connection.close()
-
+        # make sure table is deleted
+        try:
+            table = instance.table(table_id)
+            table.delete()
+        except Exception:
+            pass
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(