Commit 44926c08 authored by leswing's avatar leswing
Browse files

More flaky tests

parent baadeb61
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -141,6 +141,7 @@ class TestTensorGraph(unittest.TestCase):
      y_pred = prediction[:, i, :]
      assert_true(np.all(np.isclose(y_pred, y_real, atol=1.5)))

  @flaky
  def test_no_queue(self):
    n_data_points = 20
    n_features = 2
+5 −4
Original line number Diff line number Diff line
import numpy as np
from nose.tools import assert_true
from nose.plugins.attrib import attr
from unittest import TestCase

import deepchem as dc
import numpy as np
from deepchem.data import NumpyDataset
from deepchem.data.datasets import Databag
from deepchem.models.tensorgraph.layers import Dense, ReduceMean, SoftMax, SoftMaxCrossEntropy, L2Loss
from deepchem.models.tensorgraph.layers import Dense, ReduceMean, SoftMax, SoftMaxCrossEntropy
from deepchem.models.tensorgraph.layers import Feature, Label
from deepchem.models.tensorgraph.layers import ReduceSquareDifference
from nose.tools import assert_true
from flaky import flaky


class TestGeneratorEvaluator(TestCase):

  @flaky
  def test_compute_model_performance_multitask_classifier(self):
    n_data_points = 20
    n_features = 1