Commit 83c20492 authored by Joseph Gomes's avatar Joseph Gomes
Browse files

Added ValueError exception for joblib loader

parent a37c108a

deepchem/utils/save.py

+1 −1
Original line number Diff line number Diff line
@@ -105,7 +105,7 @@ def load_from_disk(filename): 
  elif os.path.splitext(name)[1] == ".joblib":

    try:

      return joblib.load(filename)

    except KeyError:

    except KeyError, ValueError:

      # Try older joblib version for legacy files.

      return old_joblib.load(filename)

  elif os.path.splitext(name)[1] == ".csv":

CRA Git | Maintained and supported by SUSTech CRA and CCSE