oops. bugfix

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)

minerva/redshift.py

@@ -147,6 +147,7 @@ parallel off
 manifest;
 drop table temp_data;
 """
+        print(query)
         return query