f90c78bd5444f77936566e9a67d9a7513fd3be48,tensorflow_privacy/privacy/membership_inference_attack/keras_evaluation_test.py,UtilsTest,test_run_attack_on_keras_model,#UtilsTest#,59

Before Change

        self.model,
        (self.train_data, self.train_labels),
        (self.test_data, self.test_labels),
        [])
    self.assertIsInstance(results, dict)
    self.assertIn("all_thresh_loss_auc", results)
    self.assertIn("all_thresh_loss_advantage", results)

After Change

        self.model,
        (self.train_data, self.train_labels),
        (self.test_data, self.test_labels),
        attack_types=[AttackType.THRESHOLD_ATTACK])
    self.assertIsInstance(results, AttackResults)
    attack_properties, attack_values = get_all_attack_results(results)
    self.assertLen(attack_properties, 2)
    self.assertLen(attack_values, 2)

Instances