Commit a4b6db5d authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Cleanup

parent afad4301

deepchem/trans/tests/test_balancing.py

+1 −1
Original line number Diff line number Diff line
@@ -7,7 +7,7 @@ import os 


class TestBalancingTransformer(unittest.TestCase):

  """

  Test top-level API for transformer objects.

  Test BalancingTransformer functionality. 

  """



  def test_binary_1d(self):

deepchem/trans/transformers.py

+0 −1
Original line number Diff line number Diff line
@@ -849,7 +849,6 @@ class BalancingTransformer(Transformer): 
      raise ValueError("w must be of shape (N,) or (N, n_tasks)")

    # Ensure dataset is binary

    self.classes = sorted(np.unique(y))

    #np.testing.assert_allclose(sorted(np.unique(y)), np.array([0., 1.]))

    weights = []

    for ind, task in enumerate(dataset.get_task_names()):

      task_w = w[:, ind]

CRA Git | Maintained and supported by SUSTech CRA and CCSE