diff --git a/data/maker/__init__.py b/data/maker/__init__.py
index 469a65a..967dbc8 100644
--- a/data/maker/__init__.py
+++ b/data/maker/__init__.py
@@ -28,7 +28,7 @@ def train (**args) :
     logs        = args['logs']
     real        = pd.get_dummies(df[column]).astype(np.float32).values
     labels      = pd.get_dummies(df[column_id]).astype(np.float32).values
-    
+    num_gpu     = 1 if 'num_gpu' not in args else args['num_gpu']
     max_epochs  = 10 if 'max_epochs' not in args else args['max_epochs']
     context     = args['context']
     if 'store' in args :
diff --git a/setup.py b/setup.py
index ecef3bd..aa45602 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.0.0","author":"Vanderbilt University Medical Center","author_email":"steve.l.nyemba@vanderbilt.edu","license":"MIT",
+args = {"name":"data-maker","version":"1.0.1","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.14.0','numpy==1.16.3','pandas','pandas-gbq','pymongo']
 args['url'] = 'https://hiplab.mc.vanderbilt.edu/aou/data-maker.git'