From b309916927213472f4a2d82c4d0445027caa01f2 Mon Sep 17 00:00:00 2001 From: Michael Reneer Date: Wed, 26 Jan 2022 08:40:29 -0800 Subject: [PATCH] Fix TensorFlow Privacy `g-importing-member` lint error. PiperOrigin-RevId: 424356142 --- .../privacy_tests/membership_inference_attack/models.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow_privacy/privacy/privacy_tests/membership_inference_attack/models.py b/tensorflow_privacy/privacy/privacy_tests/membership_inference_attack/models.py index 79bb826..9e78a98 100644 --- a/tensorflow_privacy/privacy/privacy_tests/membership_inference_attack/models.py +++ b/tensorflow_privacy/privacy/privacy_tests/membership_inference_attack/models.py @@ -15,7 +15,8 @@ # Lint as: python3 """Trained models for membership inference attacks.""" -from dataclasses import dataclass +import dataclasses + import numpy as np from sklearn import ensemble from sklearn import linear_model @@ -27,7 +28,7 @@ from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_s from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import DataSize -@dataclass +@dataclasses.dataclass class AttackerData: """Input data for an ML classifier attack.