From 181b8654d05da3fcea78cfadd978a28282f8736b Mon Sep 17 00:00:00 2001 From: Yu-Hua Yang <107571329+yuhuayang-google@users.noreply.github.com> Date: Fri, 1 Jul 2022 11:15:04 -0700 Subject: [PATCH] Fix: close Teradata connection via object delete (#524) --- third_party/ibis/ibis_teradata/client.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/third_party/ibis/ibis_teradata/client.py b/third_party/ibis/ibis_teradata/client.py index f3e4bee6e..fe4309b09 100644 --- 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)