fix: revert fix(presto preview): re-enable schema previsualization fo…

…r Trino/Presto table/schemas" (apache#28613)
jzhao62 · May 21, 2024 · 4fa7619 · 4fa7619
1 parent 62a0336
commit 4fa7619
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/superset/db_engine_specs/presto.py b/superset/db_engine_specs/presto.py
@@ -536,10 +536,7 @@ def where_latest_partition(
         }
 
         for col_name, value in zip(col_names, values):
-            col_type = None
-            if col_type_name := column_type_by_name.get(col_name):
-                if col_type_class := getattr(types, col_type_name, None):
-                    col_type = col_type_class()
+            col_type = column_type_by_name.get(col_name)
 
             if isinstance(col_type, types.DATE):
                 col_type = Date()

diff --git a/tests/unit_tests/db_engine_specs/test_presto.py b/tests/unit_tests/db_engine_specs/test_presto.py
@@ -116,10 +116,10 @@ def test_get_schema_from_engine_params() -> None:
 @pytest.mark.parametrize(
     ["column_type", "column_value", "expected_value"],
     [
-        ("DATE", "2023-05-01", "DATE '2023-05-01'"),
-        ("TIMESTAMP", "2023-05-01", "TIMESTAMP '2023-05-01'"),
-        ("VARCHAR", "2023-05-01", "'2023-05-01'"),
-        ("INT", 1234, "1234"),
+        (types.DATE(), "2023-05-01", "DATE '2023-05-01'"),
+        (types.TIMESTAMP(), "2023-05-01", "TIMESTAMP '2023-05-01'"),
+        (types.VARCHAR(), "2023-05-01", "'2023-05-01'"),
+        (types.INT(), 1234, "1234"),
     ],
 )
 def test_where_latest_partition(