Commit 5256cbc5 authored by Vignesh's avatar Vignesh
Browse files

Fixes to yapf and test

parent 38e20766

deepchem/molnet/load_function/bace_datasets.py

+1 −6
Original line number Diff line number Diff line
@@ -130,12 +130,7 @@ def load_bace_classification(featurizer='ECFP', 
  bace_tasks = ["Class"]



  if reload:

    save_folder = os.path.join(save_dir, "bace_c-featurized")

    if not move_mean:

      save_folder = os.path.join(save_folder, str(featurizer) + "_mean_unmoved")

    else:

      save_folder = os.path.join(save_folder, str(featurizer))



    save_folder = os.path.join(save_dir, "bace_c-featurized", str(featurizer))

    if featurizer == "smiles2img":

      img_spec = kwargs.get("img_spec", "std")

      save_folder = os.path.join(save_folder, img_spec)

deepchem/molnet/load_function/qm7_datasets.py

+1 −1
Original line number Diff line number Diff line
@@ -32,7 +32,7 @@ def load_qm7_from_mat(featurizer='CoulombMatrix', 
  if data_dir is None:

    data_dir = DEFAULT_DIR

  if save_dir is None:

    data_dir = DEFAULT_DIR

    save_dir = DEFAULT_DIR



  if reload:

    save_folder = os.path.join(save_dir, "qm7-featurized")

deepchem/molnet/load_function/sweetlead_datasets.py

+1 −1
Original line number Diff line number Diff line
@@ -13,7 +13,7 @@ import deepchem as dc 


logger = logging.getLogger(__name__)



DEFAULT_DIR = deepchem.utils.get_data_dir()

DEFAULT_DIR = dc.utils.get_data_dir()

SWEETLEAD_URL = 'http://deepchem.io.s3-website-us-west-1.amazonaws.com/datasets/sweet.csv.gz'

CRA Git | Maintained and supported by SUSTech CRA and CCSE