- working pynrrd with neuroscripts

sinkpoint · Dec 18, 2015 · 843104d · 843104d
1 parent 2f3da86
commit 843104d
Show file tree

Hide file tree

Showing 9 changed files with 3 additions and 3 deletions.
diff --git a/DESCRIPTION.rst b/DESCRIPTION.rst
diff --git a/LICENSE b/LICENSE
diff --git a/MANIFEST.in b/MANIFEST.in
diff --git a/__init__.py b/__init__.py
diff --git a/pynrrd/__init__.py b/pynrrd/__init__.py
diff --git a/pynrrd/nrrd.py b/pynrrd/nrrd.py
@@ -53,7 +53,7 @@ def getAffine(self):
 
 
     def getDwiGradients(self):
-        return self._data['DWMRI_gradient'][0]
+        return self._data['DWMRI_gradient']
 
     def setDwiGradients(self, vec):
          self._data['DWMRI_gradient'] = vec
@@ -232,7 +232,7 @@ def load(self, filename, get_raw=False):
 
         if (params.has_key(self.grdkey)):
             dwivec = params[self.grdkey]
-            params[self.grdkey] = [dwivec]
+            params[self.grdkey] = dwivec
 
         params = NrrdHeader(params)
         if not get_raw:
@@ -321,7 +321,7 @@ def write(self,nrrdheader, output):
                 line = "%s%s%s\n" % (k,eq,val)
             elif k=='DWMRI_gradient':
                 c = 0
-                for i in val[0]:
+                for i in val:
                     i = self.formatOutput(i, brac=False, dim=' ')
                     line = "%s_%04d:=%s\n" % (k, c, i)
                     c+=1

diff --git a/pynrrd/utils.py b/pynrrd/utils.py
diff --git a/setup.cfg b/setup.cfg
diff --git a/setup.py b/setup.py