ari/minerva
1
0
Fork 0
forked from bellwether/minerva
Code Pull requests Activity

oops. bugfix

Browse source
This commit is contained in:
Ari Brown 2024-03-14 09:13:59 -04:00
parent e294fc062b
commit fd289257aa
2 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
minerva/athena.py
View file

@ -227,13 +227,19 @@ class Parallelize:
def __init__(self, athena, dest=None, data=[], n=1): def __init__(self, athena, dest=None, data=[], n=1):
self.athena = athena self.athena = athena
self.dest = dest self.dest = dest
self.data = list(data)
self.n = n self.n = n
self.tables = [] self.tables = []
self.queries = [] self.queries = []
self.runtime = None self.runtime = None
self.cost = 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): def __lshift__(self, res):
self.queries.append(res) self.queries.append(res)

1
minerva/redshift.py
View file

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