Commit f1504d25 authored by leswing's avatar leswing
Browse files

Convert to list on saving

parent 8fec15be
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -471,7 +471,7 @@ class DiskDataset(Dataset):
    try:
      tasks_filename, metadata_filename = self._get_metadata_filename()
      with open(tasks_filename) as fin:
        tasks = np.array(json.load(fin))
        tasks = json.load(fin)
      metadata_df = read_hdf(metadata_filename, 'metadata')
      return tasks, metadata_df
    except Exception as e:
+3 −2
Original line number Diff line number Diff line
@@ -117,11 +117,12 @@ def save_metadata(tasks, metadata_df, data_dir):
  Returns
  -------
  """

  if isinstance(tasks, np.ndarray):
    tasks = tasks.tolist()
  metadata_filename = os.path.join(data_dir, "metadata.hd5")
  tasks_filename = os.path.join(data_dir, "tasks.json")
  with open(tasks_filename, 'w') as fout:
    json.dump(tasks.tolist(), fout)
    json.dump(tasks, fout)
  hdf = pd.HDFStore(metadata_filename)
  hdf.put('metadata', metadata_df)
  hdf.close()