瀏覽代碼

Remove depracated .value for hdfs.

cchen23 4 月之前
父節點
當前提交
6b09a20a92
共有 1 個文件被更改,包括 3 次插入3 次删除
  1. 3 3
      code/utils.py

+ 3 - 3
code/utils.py

@@ -30,9 +30,9 @@ def load_data(fname, key=None):
         if key is None:
             for k in hf.keys():
                 print("{} will be loaded".format(k))
-                data[k] = hf[k].value
+                data[k] = hf[k][()]
         else:
-            data[key] = hf[key].value
+            data[key] = hf[key][()]
     return data
 
 
@@ -89,4 +89,4 @@ def map_to_flat(voxels, mapper_file):
     mimg = (np.nan * np.ones(badmask.shape)).astype(voxels.dtype)
     mimg[badmask] = (pixmap * voxels.ravel())[badmask].astype(mimg.dtype)
     img[pixmask] = mimg
-    return img.T[::-1]
+    return img.T[::-1]