Commit a30850cc authored by nitinprakash96's avatar nitinprakash96
Browse files

fix failing test

parent ca40f1cb
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -11,6 +11,7 @@ from deepchem.data import NumpyDataset

logger = logging.getLogger(__name__)


def remove_dead_examples(dataset):
  """Removes compounds with no weight.

+1 −2
Original line number Diff line number Diff line
@@ -2,7 +2,6 @@ import os
from unittest import TestCase
from io import StringIO
import tempfile
import logging
import shutil

import deepchem as dc
@@ -14,7 +13,7 @@ class TestCSVLoader(TestCase):
    fin = tempfile.NamedTemporaryFile(mode='w', delete=False)
    fin.write("smiles,endpoint\nc1ccccc1,1")
    fin.close()
    logger.info(fin.name)
    print(fin.name)
    featurizer = dc.feat.CircularFingerprint(size=1024)
    tasks = ["endpoint"]
    loader = dc.data.CSVLoader(
+1 −0
Original line number Diff line number Diff line
@@ -14,6 +14,7 @@ import tempfile
import shutil
import deepchem as dc


class TestDataLoader(unittest.TestCase):
  """
  Test DataLoader
+1 −0
Original line number Diff line number Diff line
@@ -9,6 +9,7 @@ from sklearn.ensemble import RandomForestClassifier

logger = logging.getLogger(__name__)


class TestDrop(unittest.TestCase):
  """
  Test how loading of malformed compounds is handled.
+1 −0
Original line number Diff line number Diff line
@@ -18,6 +18,7 @@ import numpy as np

logger = logging.getLogger(__name__)


class TestLoad(unittest.TestCase):
  """
  Test singletask/multitask data loading.
Loading