From 3a6defae3911aff415584b25ca2b06519f3fb415 Mon Sep 17 00:00:00 2001
From: nicor88 <6278547+nicor88@users.noreply.github.com>
Date: Tue, 26 Sep 2023 13:19:06 +0200
Subject: [PATCH] fix issue

---
 .../materializations/models/helpers/get_partition_batches.sql | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/dbt/include/athena/macros/materializations/models/helpers/get_partition_batches.sql b/dbt/include/athena/macros/materializations/models/helpers/get_partition_batches.sql
index b9dfebb9..3f64cc59 100644
--- a/dbt/include/athena/macros/materializations/models/helpers/get_partition_batches.sql
+++ b/dbt/include/athena/macros/materializations/models/helpers/get_partition_batches.sql
@@ -28,7 +28,7 @@
             {%- if col is none -%}
                 {%- set value = 'null' -%}
                 {%- set comp_func = ' is ' -%}
-            {%- elif column_type == 'integer' -%}
+            {%- elif column_type == 'integer' or column_type is none -%}
                 {%- set value = col | string -%}
             {%- elif column_type == 'string' -%}
                 {%- set value = "'" + col + "'" -%}
@@ -36,8 +36,6 @@
                 {%- set value = "DATE'" + col | string + "'" -%}
             {%- elif column_type == 'timestamp' -%}
                 {%- set value = "TIMESTAMP'" + col | string + "'" -%}
-            {%- elif column_type is none -%}
-                {%- set value = col | string -%}
             {%- else -%}
                 {%- do exceptions.raise_compiler_error('Need to add support for column type ' + column_type) -%}
             {%- endif -%}