5b62ac044be8353759b4ce159f507e5f2b985157,tests/test_track.py,,test_track_ensemble,#Any#Any#,112

Before Change

    temp_dir = tempfile.TemporaryDirectory()
    recon_path = temp_dir.name + "/model_file.npy"
    model, _ = track.reconstruction(conn_model, gtab, dwi_data, wm_in_dwi)
    np.save(recon_path, model)

    streamlines = track.track_ensemble(target_samples, atlas_data_wm_gm_int,
                                       labels_im_file,
                   recon_path, sphere, directget, curv_thr_list, step_list,

After Change

    recon_path = temp_dir.name + "/model_file.hdf5"
    model, _ = track.reconstruction(conn_model, gtab, dwi_data, wm_in_dwi)

    with h5py.File(recon_path, "w") as hf:
        hf.create_dataset("reconstruction",
                          data=model.astype("float32"))
    hf.close()

Instances