From c852e2bd373b427568f9648e048672a9f2f983cf Mon Sep 17 00:00:00 2001 From: Maya Anderson Date: Wed, 20 Sep 2023 20:11:10 +0300 Subject: [PATCH] Fix imports Signed-off-by: Maya Anderson --- apt/risk/data_assessment/dataset_assessment_manager.py | 4 ++-- tests/test_data_assessment.py | 6 +++--- tests/test_data_assessment_short_test.py | 9 +++++---- 3 files changed, 10 insertions(+), 9 deletions(-) diff --git a/apt/risk/data_assessment/dataset_assessment_manager.py b/apt/risk/data_assessment/dataset_assessment_manager.py index 8924f0d..4d699bc 100644 --- a/apt/risk/data_assessment/dataset_assessment_manager.py +++ b/apt/risk/data_assessment/dataset_assessment_manager.py @@ -13,8 +13,8 @@ from apt.risk.data_assessment.dataset_attack_result import DatasetAttackScore, D from apt.risk.data_assessment.dataset_attack_whole_dataset_knn_distance import \ DatasetAttackConfigWholeDatasetKnnDistance, DatasetAttackWholeDatasetKnnDistance from apt.utils.datasets import ArrayDataset -from data_assessment.dataset_attack_membership_classification import DatasetAttackConfigMembershipClassification, \ - DatasetAttackMembershipClassification +from apt.risk.data_assessment.dataset_attack_membership_classification import \ + DatasetAttackConfigMembershipClassification, DatasetAttackMembershipClassification @dataclass diff --git a/tests/test_data_assessment.py b/tests/test_data_assessment.py index ae32e19..93d595a 100644 --- a/tests/test_data_assessment.py +++ b/tests/test_data_assessment.py @@ -13,9 +13,9 @@ from apt.risk.data_assessment.dataset_assessment_manager import DatasetAssessmen from apt.utils.dataset_utils import get_iris_dataset_np, get_diabetes_dataset_np, get_adult_dataset_pd, \ get_nursery_dataset_pd from apt.utils.datasets import ArrayDataset -from data_assessment.dataset_attack_membership_classification import DatasetAttackScoreMembershipClassification -from data_assessment.dataset_attack_membership_knn_probabilities import DatasetAttackScoreMembershipKnnProbabilities -from data_assessment.dataset_attack_whole_dataset_knn_distance import DatasetAttackScoreWholeDatasetKnnDistance +from apt.risk.data_assessment.dataset_attack_membership_classification import DatasetAttackScoreMembershipClassification +from apt.risk.data_assessment.dataset_attack_membership_knn_probabilities import DatasetAttackScoreMembershipKnnProbabilities +from apt.risk.data_assessment.dataset_attack_whole_dataset_knn_distance import DatasetAttackScoreWholeDatasetKnnDistance MIN_SHARE = 0.5 MIN_ROC_AUC = 0.0 diff --git a/tests/test_data_assessment_short_test.py b/tests/test_data_assessment_short_test.py index d674685..7416d47 100644 --- a/tests/test_data_assessment_short_test.py +++ b/tests/test_data_assessment_short_test.py @@ -4,10 +4,11 @@ from apt.anonymization import Anonymize from apt.risk.data_assessment.dataset_assessment_manager import DatasetAssessmentManager, DatasetAssessmentManagerConfig from apt.utils.dataset_utils import get_iris_dataset_np, get_nursery_dataset_pd from apt.utils.datasets import ArrayDataset -from data_assessment.dataset_attack_membership_classification import DatasetAttackScoreMembershipClassification -from data_assessment.dataset_attack_membership_knn_probabilities import DatasetAttackScoreMembershipKnnProbabilities, \ - DatasetAttackConfigMembershipKnnProbabilities, DatasetAttackMembershipKnnProbabilities -from data_assessment.dataset_attack_whole_dataset_knn_distance import DatasetAttackScoreWholeDatasetKnnDistance +from apt.risk.data_assessment.dataset_attack_membership_classification import DatasetAttackScoreMembershipClassification +from apt.risk.data_assessment.dataset_attack_membership_knn_probabilities import \ + DatasetAttackScoreMembershipKnnProbabilities, DatasetAttackConfigMembershipKnnProbabilities, \ + DatasetAttackMembershipKnnProbabilities +from apt.risk.data_assessment.dataset_attack_whole_dataset_knn_distance import DatasetAttackScoreWholeDatasetKnnDistance from tests.test_data_assessment import kde, preprocess_nursery_x_data NUM_SYNTH_SAMPLES = 10