Fix bug (e17ce1a9) · Commits · 钟慕尧 / deepchem

deepchem/models/test/test_singletask_to_multitask.py

+2 −2

Original line number	Diff line number	Diff line
		@@ -42,8 +42,8 @@ class TestSingletasktoMultitaskAPI(TestAPI):
		"data_shape": train_dataset.get_data_shape()
		}
		classification_metrics = [Metric(metrics.roc_auc_score)]
		def model_builder(task_types, model_params, verbosity=None):
		return SklearnModel(task_types, model_params,
		def model_builder(task_types, model_params, model_builder, verbosity=None):
		return SklearnModel(task_types, model_params, model_builder,
		model_instance=LogisticRegression())
		multitask_model = SingletaskToMultitask(task_types, params_dict,
		self.model_dir, model_builder)