Fix: close Teradata connection via object delete (#524)

GoogleCloudPlatform · Jul 1, 2022 · 181b865 · 181b865
1 parent e87d0ef
commit 181b865
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/third_party/ibis/ibis_teradata/client.py b/third_party/ibis/ibis_teradata/client.py
@@ -93,6 +93,9 @@ def __init__(self, host, user_name, password, port=1025, logmech='TD2', use_no_l
         self.client = teradatasql.connect(**self.teradata_config)
         self.use_no_lock_tables = use_no_lock_tables
 
+    def __del__ (self):
+        teradatasql.close()
+
     def _execute(self, dml, results=False, **kwargs):
         query = TeradataQuery(self, dml)
         df = self._execute_query(query, **kwargs)