From fd289257aa5909461dfd2b19e04ddbec067fcc9c Mon Sep 17 00:00:00 2001
From: Ari Brown <ari@airintech.com>
Date: Thu, 14 Mar 2024 09:13:59 -0400
Subject: [PATCH] oops. bugfix

---
 minerva/athena.py   | 8 +++++++-
 minerva/redshift.py | 1 +
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/minerva/athena.py b/minerva/athena.py
index 4423bff..0a316e4 100644
--- a/minerva/athena.py
+++ b/minerva/athena.py
@@ -227,13 +227,19 @@ class Parallelize:
     def __init__(self, athena, dest=None, data=[], n=1):
         self.athena  = athena
         self.dest    = dest
-        self.data    = list(data)
         self.n       = n
         self.tables  = []
         self.queries = []
         self.runtime = None
         self.cost    = None
 
+        if type(data) == type(1):
+            self.data = list(range(data))
+        elif type(data) == type([]) or data.__iter__:
+            self.data = list(data)
+        else:
+            raise Exception(f"Passed in {type(data)}, expected list-like or integer")
+
 
     def __lshift__(self, res):
         self.queries.append(res)
diff --git a/minerva/redshift.py b/minerva/redshift.py
index 697107e..2d7b1c1 100644
--- a/minerva/redshift.py
+++ b/minerva/redshift.py
@@ -147,6 +147,7 @@ parallel off
 manifest;
 drop table temp_data;
 """
+        print(query)
         return query