fix: Adding Decimal datatype support for MSSQL custom query validation (

#771) * fix: Adding Decimal datatype support for MSSQL custom query validation * Adding only Decimal mapping for IBIS --------- Co-authored-by: Piyush Sarraf <[email protected]>
GoogleCloudPlatform · Mar 23, 2023 · 0d5c5eb · 0d5c5eb
1 parent 89413c1
commit 0d5c5eb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/third_party/ibis/ibis_mssql/client.py b/third_party/ibis/ibis_mssql/client.py
@@ -15,6 +15,7 @@
 
 import contextlib
 import datetime
+import decimal
 import getpass
 
 import sqlalchemy as sa
@@ -256,6 +257,7 @@ def _get_schema_using_query(self, limited_query):
             float: 'float64',
             str: 'string',
             datetime.datetime: 'timestamp',
+            decimal.Decimal: 'Decimal'
         }
 
         with self._execute(limited_query, results=True) as cur: