Merge pull request #973 from kamran-haider/fix_featurizer_name_error (0d42aef7) · Commits · 钟慕尧 / deepchem

examples/benchmark_curve.py

+8 −0

Original line number	Diff line number	Diff line
		@@ -104,6 +104,14 @@ for dataset in datasets:
		if pair in CheckFeaturizer:
		featurizer = CheckFeaturizer[pair][0]
		n_features = CheckFeaturizer[pair][1]
		else:
		supported_combinations = [
		key for key in CheckFeaturizer.keys() if pair[0] == key[0]
		]
		supported_models = [k[1] for k in supported_combinations]
		raise ValueError(
		"Model %s not supported for %s dataset. Please choose from the following:\n%s"
		% (pair[1], pair[0], " ".join(supported_models)))

		tasks, all_dataset, transformers = load_dataset(
		dataset, featurizer, split='index')