Fix yapf identified formatting issues (b97e1f72) · Commits · 钟慕尧 / deepchem

deepchem/models/tensorgraph/models/graph_models.py

+4 −2

Original line number	Diff line number	Diff line
		@@ -1149,11 +1149,13 @@ class MPNNTensorGraph(TensorGraph):

		import warnings


		class GraphConvTensorGraph(GraphConvModel):

		warnings.warn("GraphConvTensorGraph is deprecated and has been renamed to GraphConvModel and will be removed in DeepChem 3.0.", FutureWarning)
		warnings.warn(
		"GraphConvTensorGraph is deprecated and has been renamed to GraphConvModel and will be removed in DeepChem 3.0.",
		FutureWarning)

		def __init__(self, args, *kwargs):

		super(GraphConvTensorGraph, self).__init__(args, *kwargs)

+3 −6

Original line number	Diff line number	Diff line
		@@ -58,8 +58,7 @@ class TestGraphModels(unittest.TestCase):
		'regression', 'GraphConv')

		batch_size = 50
		model = GraphConvModel(
		len(tasks), batch_size=batch_size, mode='regression')
		model = GraphConvModel(len(tasks), batch_size=batch_size, mode='regression')

		model.fit(dataset, nb_epoch=1)
		scores = model.evaluate(dataset, [metric], transformers)
		@@ -73,8 +72,7 @@ class TestGraphModels(unittest.TestCase):
		'regression', 'GraphConv', num_tasks=1)

		batch_size = 50
		model = GraphConvModel(
		len(tasks), batch_size=batch_size, mode='regression')
		model = GraphConvModel(len(tasks), batch_size=batch_size, mode='regression')

		model.fit(dataset, nb_epoch=1)

		@@ -121,8 +119,7 @@ class TestGraphModels(unittest.TestCase):
		'regression', 'GraphConv', num_tasks=1)

		batch_size = 50
		model = GraphConvModel(
		len(tasks), batch_size=batch_size, mode='regression')
		model = GraphConvModel(len(tasks), batch_size=batch_size, mode='regression')

		model.fit(dataset, nb_epoch=1)
		model.save()

+1 −2

Original line number	Diff line number	Diff line
		@@ -74,8 +74,7 @@ def experiment(dataset_file, method='GraphConv', split='scaffold'):

		model = None
		if method == 'GraphConv':
		model = GraphConvModel(
		len(tasks), batch_size=BATCH_SIZE, mode="regression")
		model = GraphConvModel(len(tasks), batch_size=BATCH_SIZE, mode="regression")
		elif method == 'RF':

		def model_builder_rf(model_dir):