Commit 5b5ba9c2 authored by alat-rights's avatar alat-rights
Browse files

cleanup

parent b0416862

deepchem/data/data_loader.py

+0 −1
Original line number Diff line number Diff line
@@ -990,7 +990,6 @@ class FASTALoader(DataLoader): 
      for input_file in input_files:

        sequences = np.append(sequences, _read_file(input_file))

      X = self.featurizer(sequences)

      logger.warning(X)

      ids = np.ones(len(X))

      # (X, y, w, ids)

      yield X, None, None, ids

deepchem/data/tests/test_fasta_loader.py

+1 −3
Original line number Diff line number Diff line
@@ -30,7 +30,6 @@ class TestFASTALoader(unittest.TestCase): 
    if legacy:

      assert sequences.X.shape == (3, 5, 58, 1)

    else:

      assert False

      assert sequences.X.shape == (3, 58, 5)



  def test_fasta_one_hot_big(self):
@@ -39,5 +38,4 @@ class TestFASTALoader(unittest.TestCase): 
    loader = dc.data.FASTALoader(charset="protein", max_length = 1000)

    sequences = loader.create_dataset(input_file)



    print(sequences.X.shape)

    assert False
 
    assert sequences.X.shape

CRA Git | Maintained and supported by SUSTech CRA and CCSE