Commit 4ec00865 authored by Peter Eastman's avatar Peter Eastman
Browse files

write_data_to_disk() tried to compute statistics even with compute_feature_statistics=False

parent 5426829b

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