From 2b7b1757f92840a65fa8a599ff475f06e7be00ca Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Wed, 4 Mar 2020 14:17:28 -0600 Subject: [PATCH] bug fixes with callback (worker) --- pipeline.py | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pipeline.py b/pipeline.py index b53ba52..a39dbc7 100644 --- a/pipeline.py +++ b/pipeline.py @@ -211,7 +211,7 @@ class Components : args = info['args'] args['gpu'] = int(info['info']['partition']) if int(args['num_gpu']) > 1 and args['gpu'] > 0: - args['gpu'] = args['gpu'] + args['num_gpu'] if args['gpu'] + args['num_gpu'] < 8 else 0 + args['gpu'] = args['gpu'] + args['num_gpu'] if args['gpu'] + args['num_gpu'] < 8 else 0 #-- 8 max gpus args['reader'] = lambda: df # # @TODO: Fix diff --git a/setup.py b/setup.py index e6b988a..26e2e9d 100644 --- a/setup.py +++ b/setup.py @@ -4,7 +4,7 @@ import sys def read(fname): return open(os.path.join(os.path.dirname(__file__), fname)).read() -args = {"name":"data-maker","version":"1.1.6","author":"Vanderbilt University Medical Center","author_email":"steve.l.nyemba@vanderbilt.edu","license":"MIT", +args = {"name":"data-maker","version":"1.1.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'] args['url'] = 'https://hiplab.mc.vanderbilt.edu/git/aou/data-maker.git'