66c20cc2157d7f18530c742ce6d71685f8699020,tests/classifiers/test_tensorflow.py,TestTensorFlowClassifier,setUpClass,#Any#,46

Before Change

    def setUpClass(cls):
        (x_train, y_train), (x_test, y_test), _, _ = load_dataset("mnist")

        cls.x_train = x_train[:NB_TRAIN]
        cls.y_train = y_train[:NB_TRAIN]
        cls.x_test = x_test[:NB_TEST]
        cls.y_test = y_test[:NB_TEST]

        cls.classifier, cls.sess = get_classifier_tf()
        cls.classifier_logits, _ = get_classifier_tf(from_logits=True)

After Change

    @classmethod
    def setUpClass(cls):
        super().setUpClass()

        cls.classifier, cls.sess = get_classifier_tf()
        cls.classifier_logits, _ = get_classifier_tf(from_logits=True)

Instances