Commit 2753d5ff authored by peastman's avatar peastman
Browse files

Fixed error on Python 2

parent 354ccf61

deepchem/models/tensorgraph/tests/test_seqtoseq.py

+1 −1
Original line number Diff line number Diff line
@@ -37,7 +37,7 @@ class TestSeqToSeq(unittest.TestCase): 


    # Test it out.



    tests = [s for s, t in generate_sequences(sequence_length, 50)]

    tests = [seq for seq, target in generate_sequences(sequence_length, 50)]

    pred1 = s.predict_from_sequences(tests, beam_width=1)

    pred4 = s.predict_from_sequences(tests, beam_width=4)

    embeddings = s.predict_embeddings(tests)

CRA Git | Maintained and supported by SUSTech CRA and CCSE