Attempting to add fixed graph to prevent test runner issues (f8a1e468) · Commits · 钟慕尧 / deepchem

deepchem/models/tests/test_overfit.py

+128 −120

Original line number	Diff line number	Diff line
		@@ -668,6 +668,10 @@ class TestOverfitAPI(test_util.TensorFlowTestCase):

		def test_graph_conv_multitask_classification_overfit(self):
		"""Test graph-conv multitask overfits tiny data."""
		g = tf.Graph()
		sess = tf.Session(graph=g)
		K.set_session(sess)
		with g.as_default():
		n_tasks = 1
		n_samples = 10
		n_features = 3
		@@ -720,6 +724,10 @@ class TestOverfitAPI(test_util.TensorFlowTestCase):

		def test_attn_lstm_multitask_classification_overfit(self):
		"""Test support graph-conv multitask overfits tiny data."""
		g = tf.Graph()
		sess = tf.Session(graph=g)
		K.set_session(sess)
		with g.as_default():
		n_tasks = 1
		n_feat = 71
		max_depth = 4