Commit b74bbade authored by leswing's avatar leswing
Browse files

no index in the csv file

parent 8a1a2a02

deepchem/data/datasets.py

+1 −0
Original line number Diff line number Diff line
@@ -486,6 +486,7 @@ class DiskDataset(Dataset): 
      with open(tasks_filename) as fin:

        tasks = json.load(fin)

      metadata_df = pd.read_csv(metadata_filename, compression='gzip')

      metadata_df = metadata_df.where((pd.notnull(metadata_df)), None)

      return tasks, metadata_df

    except Exception as e:

      pass

deepchem/utils/save.py

+1 −1
Original line number Diff line number Diff line
@@ -123,7 +123,7 @@ def save_metadata(tasks, metadata_df, data_dir): 
  tasks_filename = os.path.join(data_dir, "tasks.json")

  with open(tasks_filename, 'w') as fout:

    json.dump(tasks, fout)

  metadata_df.to_csv(metadata_filename, compression='gzip')

  metadata_df.to_csv(metadata_filename, index=False, compression='gzip')





def load_from_disk(filename):

CRA Git | Maintained and supported by SUSTech CRA and CCSE