yapf (again) (62dd4d94) · Commits · 钟慕尧 / deepchem

deepchem/models/tensorgraph/tests/test_layers.py

+9 −10

Original line number	Diff line number	Diff line
		@@ -270,8 +270,8 @@ class TestLayers(test_util.TensorFlowTestCase):
		coords_tensor = tf.convert_to_tensor(coords)
		nbr_list_tensor = tf.convert_to_tensor(nbr_list)

		dist_tensor = InteratomicL2Distances(N_atoms, M_nbrs, ndim)(
		coords_tensor, nbr_list_tensor)
		dist_tensor = InteratomicL2Distances(N_atoms, M_nbrs,
		ndim)(coords_tensor, nbr_list_tensor)

		dists = dist_tensor.eval()
		assert dists.shape == (N_atoms, M_nbrs)
		@@ -419,8 +419,7 @@ class TestLayers(test_util.TensorFlowTestCase):
		lstm = LSTMStep(n_feat, 2 * n_feat)
		out_tensor = lstm(y, state_zero, state_one)
		sess.run(tf.global_variables_initializer())
		h_out, h_copy_out, c_out = (out_tensor[0].eval(),
		out_tensor[1][0].eval(),
		h_out, h_copy_out, c_out = (out_tensor[0].eval(), out_tensor[1][0].eval(),
		out_tensor[1][1].eval())
		assert h_out.shape == (n_test, n_feat)
		assert h_copy_out.shape == (n_test, n_feat)
		@@ -460,8 +459,8 @@ class TestLayers(test_util.TensorFlowTestCase):
		test = tf.convert_to_tensor(test, dtype=tf.float32)
		support = tf.convert_to_tensor(support, dtype=tf.float32)

		iter_ref_embedding_layer = IterRefLSTMEmbedding(n_test, n_support,
		n_feat, max_depth)
		iter_ref_embedding_layer = IterRefLSTMEmbedding(n_test, n_support, n_feat,
		max_depth)
		out_tensor = iter_ref_embedding_layer(test, support)
		sess.run(tf.global_variables_initializer())
		test_out, support_out = out_tensor[0].eval(), out_tensor[1].eval()