Commit 139b7ae5 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Fixes for broken tests

parent e0e475c5

deepchem/hyperparameters/tests/test_hyperparam_opt.py

+1 −1
Original line number Diff line number Diff line
@@ -22,7 +22,7 @@ from deepchem import metrics 
from deepchem.metrics import Metric

from sklearn.ensemble import RandomForestRegressor



def rf_model_builder(task_types, params_dict, logdir=None):

def rf_model_builder(task_types, params_dict, logdir=None, verbosity=None):

    """Builds random forests given hyperparameters.



    Last two arguments only for tensorflow models and ignored.

deepchem/models/tensorflow_models/__init__.py

+1 −1
Original line number Diff line number Diff line
@@ -331,7 +331,7 @@ class TensorflowGraph(object): 
        logging.info('Eval batch took %g seconds', time.time() - start)



        labels = np.array(from_one_hot(

            np.squeeze(np.concatenate(labels)), axis=2))

            np.squeeze(np.concatenate(labels)), axis=-1))



    return np.copy(labels)

deepchem/models/test/test_tf_api.py

+1 −0
Original line number Diff line number Diff line
@@ -69,6 +69,7 @@ class TestTensorflowAPI(TestAPI): 
      "bias_init_consts": [0.],

      "dropouts": [.5],

      "num_classes": 2,

      "nb_epoch": 1,

      "penalty": 0.0,

      "optimizer": "adam",

      "learning_rate": .001,

CRA Git | Maintained and supported by SUSTech CRA and CCSE