# HG changeset patch
# User greg
# Date 1509888894 18000
# Node ID 88ff4adbfb2fe9c73a3cd6b5205c0610aee4ed3d
# Parent  0c3113fdfae6d0009aa02fff43ccd69941373652
Uploaded

diff -r 0c3113fdfae6 -r 88ff4adbfb2f dmri.py
--- a/dmri.py	Sun Nov 05 07:43:52 2017 -0500
+++ b/dmri.py	Sun Nov 05 08:34:54 2017 -0500
@@ -18,12 +18,12 @@
 
 # Get input data.
 input_dir = args.drmi_dataset
-if args.drmi_dataset == 'sherbrooke_3shell':
+if input_dir == 'sherbrooke_3shell':
     fetch_sherbrooke_3shell()
     fdwi = os.path.join(input_dir, 'HARDI193.nii.gz')
     fbval = os.path.join(input_dir, 'HARDI193.bval')
     fbvec = os.path.join(input_dir, 'HARDI193.bvec')
-elif args.drmi_dataset == 'stanford_hardi':
+elif input_dir == 'stanford_hardi':
     fetch_stanford_hardi()
     fdwi = os.path.join(input_dir, 'HARDI150.nii.gz')
     fbval = os.path.join(input_dir, 'HARDI150.bval')