Commit a94dcf62 authored by leswing's avatar leswing
Browse files

Convert to list after type check

parent f1504d25

deepchem/utils/save.py

+1 −0
Original line number Diff line number Diff line
@@ -117,6 +117,7 @@ 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")

CRA Git | Maintained and supported by SUSTech CRA and CCSE