eb668c03c6a92d0ee2f7edbd46417fb0f7b85ae5,art/detection/features_unittest.py,TestFeatures,setUpClass,#Any#,26

Before Change

    @classmethod
    def setUpClass(cls):
        dir_path = "./tests"
        if not os.path.exists(dir_path):
            os.makedirs(dir_path)

        // Initialize a tf session
        session = tf.Session()
        k.set_session(session)

        // Get MNIST

After Change

        (x_train, y_train), (x_test, y_test), _, _ = load_mnist()
        x_train, y_train = x_train[:nb_train], y_train[:nb_train]
        x_test, y_test = x_test[:NB_TEST], y_test[:NB_TEST]
        cls.mnist = (x_train, y_train), (x_test, y_test)

        input_shape = x_train.shape[1:]
        nb_classes = 10

Instances