Commit cbaa4c3a authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #312 from peastman/stats 

write_data_to_disk() tried to compute statistics even with compute_feature_statistics=False
parents 5426829b 4ec00865

deepchem/data/datasets.py

+5 −4
Original line number Diff line number Diff line
@@ -469,6 +469,7 @@ class DiskDataset(Dataset): 
    if y is not None:

      save_to_disk(y, os.path.join(data_dir, out_y))

      save_to_disk(y, os.path.join(data_dir, out_y_transformed))

      if compute_feature_statistics:

        y_sums, y_sum_squares, y_n = compute_sums_and_nb_sample(y, w)

        save_to_disk(y_sums, os.path.join(data_dir, out_y_sums))

        save_to_disk(y_sum_squares, os.path.join(data_dir, out_y_sum_squares))

CRA Git | Maintained and supported by SUSTech CRA and CCSE