# HG changeset patch
# User greg
# Date 1512057061 18000
# Node ID d9c4ef2ee7c6f4217f3df4cd9c0d2c0525ff2ef2
# Parent  501221517d3f27e69273b4875a4589d6b2bf2bfb
Uploaded

diff -r 501221517d3f -r d9c4ef2ee7c6 dmri.py
--- a/dmri.py	Thu Nov 30 10:47:25 2017 -0500
+++ b/dmri.py	Thu Nov 30 10:51:01 2017 -0500
@@ -34,8 +34,8 @@
         os.rmdir(source_directory)
 
 # Get input data.
-input_dir = args.drmi_dataset
-if input_dir == 'stanford_hardi':
+input_dir = 'stanford_hardi'
+if args.drmi_dataset == 'stanford_hardi':
     if args.drmi_dataset_type == "dataset":
         fetch_stanford_hardi()
         fdwi = os.path.join(input_dir, 'HARDI150.nii.gz')
@@ -48,7 +48,7 @@
         fbval = os.path.join(input_dir, 'HARDI150.bval')
         fbvec = os.path.join(input_dir, 'HARDI150.bvec')
 else:
-    # input_dir == 'stanford_t1
+    # args.drmi_dataset == 'stanford_t1
         fetch_stanford_t1()
         fdwi = os.path.join(input_dir, 't1.nii.gz')
         fbval = os.path.join(input_dir, 't1.bval')