Fix trino temp tables cleanup

elementary-data · Jul 8, 2024 · eedf5d1 · eedf5d1
1 parent 8572db5
commit eedf5d1
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/macros/edr/tests/test_utils/clean_up_tables.sql b/macros/edr/tests/test_utils/clean_up_tables.sql
@@ -10,4 +10,10 @@
     {% for relation in relations %}
         {% do adapter.clean_up_table(relation) %}
     {% endfor %}
-{% endmacro %}
+{% endmacro %}
+
+{% macro trino__clean_up_tables(relations) %}
+    {% for relation in relations %}
+        {% do adapter.clean_up_table(relation) %}
+    {% endfor %}
+{% endmacro %}
diff --git a/macros/utils/table_operations/fully_drop_relation.sql b/macros/utils/table_operations/fully_drop_relation.sql
@@ -9,4 +9,9 @@
 {% macro athena__fully_drop_relation(relation) %}
     {% do adapter.clean_up_table(relation) %}
     {% do adapter.drop_relation(relation) %}
-{% endmacro %}
+{% endmacro %}
+
+{% macro trino__fully_drop_relation(relation) %}
+    {% do adapter.clean_up_table(relation) %}
+    {% do adapter.drop_relation(relation) %}
+{% endmacro %}