Unverified Commit c288fd5a authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #1480 from VIGS25/move-tests 

Moved some tests from tensorgraph.models to tensorgraph.tests
parents be8ce516 3073c288

deepchem/models/tensorgraph/tests/test_estimators.py

+2 −2
Original line number Diff line number Diff line
@@ -460,8 +460,8 @@ class TestEstimators(unittest.TestCase): 
  @flaky

  def test_dtnn_regression_model(self):

    """Test creating an estimator for DTNNGraphModel for regression"""

    data_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "../")

    input_file = os.path.join(data_dir, "models", "example_DTNN.mat")

    current_dir = os.path.dirname(os.path.abspath(__file__))

    input_file = os.path.join(current_dir, "example_DTNN.mat")

    dataset = loadmat(input_file)



    np.random.seed(123)

CRA Git | Maintained and supported by SUSTech CRA and CCSE