fix: Fix sql for missing columns and mysql based dialects (amundsen-i…

…o#550) (amundsen-io#305) This solves amundsen-io#550
DataChefHQ · Jun 30, 2022 · 784056d · 784056d
1 parent 1a906c8
commit 784056d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/databuilder/example/dags/hive_sample_dag.py b/databuilder/example/dags/hive_sample_dag.py
@@ -66,7 +66,8 @@ def connection_string():
 def create_table_wm_job(**kwargs):
     sql = textwrap.dedent("""
         SELECT From_unixtime(A0.create_time) as create_time,
-               C0.NAME as schema,
+               'hive'                        as `database`,
+               C0.NAME                       as `schema`,
                B0.tbl_name as table_name,
                {func}(A0.part_name) as part_name,
                {watermark} as part_type