diff --git a/pipeline.py b/pipeline.py index 5759696..2311007 100644 --- a/pipeline.py +++ b/pipeline.py @@ -244,7 +244,7 @@ class Components : writer = factory.instance(**_args['store']) _df = _args['data'] if _schema : - columns = [] + columns = _df.columns.tolist for _item in _schema : name = _item['name'] _type = str @@ -266,7 +266,7 @@ class Components : columns.append(name) writer.write(_df,schema=_schema,table=args['from']) else: - writer.write(_df[columns],table=args['from']) + writer.write(_df,table=args['from']) def finalize(self,args): """ diff --git a/setup.py b/setup.py index d75f1d3..1efc05e 100644 --- a/setup.py +++ b/setup.py @@ -5,7 +5,7 @@ import sys def read(fname): return open(os.path.join(os.path.dirname(__file__), fname)).read() args = {"name":"data-maker", - "version":"1.4.6", + "version":"1.4.7", "author":"Vanderbilt University Medical Center","author_email":"steve.l.nyemba@vanderbilt.edu","license":"MIT", "packages":find_packages(),"keywords":["healthcare","data","transport","protocol"]} args["install_requires"] = ['data-transport@git+https://dev.the-phi.com/git/steve/data-transport.git','tensorflow==1.15','pandas','pandas-gbq','pymongo']