Fix: dialect build

tobymao · Jun 15, 2023 · b86f7e8 · b86f7e8
1 parent 9660c33
commit b86f7e8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/sqlglot/expressions.py b/sqlglot/expressions.py
@@ -3379,8 +3379,9 @@ def build(
         from sqlglot import parse_one
 
         if isinstance(dtype, str):
-            if dtype.upper() in cls.Type.__members__:
-                data_type_exp: t.Optional[Expression] = DataType(this=DataType.Type[dtype.upper()])
+            dtype = dtype.upper()
+            if not dialect and dtype in cls.Type.__members__:
+                data_type_exp: t.Optional[Expression] = DataType(this=DataType.Type[dtype])
             else:
                 data_type_exp = parse_one(dtype, read=dialect, into=DataType)
 

diff --git a/tests/test_expressions.py b/tests/test_expressions.py
@@ -818,6 +818,7 @@ def test_data_type_builder(self):
         self.assertEqual(exp.DataType.build("OBJECT").sql(), "OBJECT")
         self.assertEqual(exp.DataType.build("NULL").sql(), "NULL")
         self.assertEqual(exp.DataType.build("UNKNOWN").sql(), "UNKNOWN")
+        self.assertEqual(exp.DataType.build("TIMESTAMP", dialect="bigquery").sql(), "TIMESTAMPTZ")
 
     def test_rename_table(self):
         self.assertEqual(