From 6b7404a8ec29741b297b132748f5c9cd5330ded5 Mon Sep 17 00:00:00 2001
From: Daniel Sanche <d.sanche14@gmail.com>
Date: Thu, 14 Dec 2023 14:57:30 -0800
Subject: [PATCH] added line to make sure tables are removed

---
 samples/hello_happybase/main.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/samples/hello_happybase/main.py b/samples/hello_happybase/main.py
index 7999fd006..698541617 100644
--- 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(