Commit 84e20a7f authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

test fixes

parent 1e125d40
Loading
Loading
Loading
Loading
+7 −5
Original line number Diff line number Diff line
@@ -7,7 +7,6 @@ from __future__ import unicode_literals
from deepchem.molnet import load_delaney
from deepchem.trans.transformers import FeaturizationTransformer
from deepchem.trans.transformers import DataTransforms
from tensorflow.examples.tutorials.mnist import input_data

__author__ = "Bharath Ramsundar"
__copyright__ = "Copyright 2016, Stanford University"
@@ -18,6 +17,7 @@ import unittest
import numpy as np
import pandas as pd
import deepchem as dc
import tensorflow as tf
import scipy.ndimage
from PIL import Image

@@ -33,12 +33,14 @@ class TestTransformers(unittest.TestCase):
    '''
       init to load the MNIST data for DataTransforms Tests
      '''
    mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)
    #mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)
    (x_train, y_train), (x_test, y_test) = tf.keras.datasets.mnist.load_data()
    # extracting validation set of MNIST for testing the DataTransforms
    valid = dc.data.NumpyDataset(mnist.validation.images,
                                 mnist.validation.labels)
    #valid = dc.data.NumpyDataset(mnist.validation.images,
    #                             mnist.validation.labels)
    train = dc.data.NumpyDataset(x_train, y_train)
    # extract only the images (no need of the labels)
    data = (valid.X)[0]
    data = (train.X)[0]
    # reshaping the vector to image
    data = np.reshape(data, (28, 28))
    self.d = data
+2 −0
Original line number Diff line number Diff line
@@ -56,6 +56,8 @@ conda install -y -q -c deepchem -c rdkit -c conda-forge -c omnia \
    zlib \
    requests \
    xgboost \
    biopython \
    pillow \
    setuptools \
    numpy
yes | pip install $tensorflow==2.1.0 tensorflow-probability