Commit 8fec15be authored by leswing's avatar leswing
Browse files

numpy vs list

parent 3819b925
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 = json.load(fin)
        tasks = np.array(json.load(fin))
      metadata_df = read_hdf(metadata_filename, 'metadata')
      return tasks, metadata_df
    except Exception as e:
+1 −1
Original line number Diff line number Diff line
@@ -121,7 +121,7 @@ def save_metadata(tasks, metadata_df, data_dir):
  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, fout)
    json.dump(tasks.tolist(), fout)
  hdf = pd.HDFStore(metadata_filename)
  hdf.put('metadata', metadata_df)
  hdf.close()