Remove unused imports

pps-lab · Feb 27, 2023 · c0b799f · c0b799f
1 parent 9d602fb
commit c0b799f
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 5 deletions.
diff --git a/src/attack/__init__.py b/src/attack/__init__.py
@@ -1,5 +1,4 @@
 
 from .targeted_attack import TargetedAttack
 from .untargeted_attack import UntargetedAttack
-from .anticipate_attack import AnticipateAttack
 from .anticipate_tf_attack import AnticipateTfAttack
diff --git a/src/attack/anticipate_tf_attack.py b/src/attack/anticipate_tf_attack.py
@@ -8,7 +8,7 @@
 logger = logging.getLogger(__name__)
 
 # Move this into generate later
-from src.torch_compat.anticipate import train_anticipate
+# from src.torch_compat.anticipate import train_anticipate
 
 class AnticipateTfAttack(LossBasedAttack):
 

diff --git a/src/federated_averaging.py b/src/federated_averaging.py
@@ -12,7 +12,7 @@
 import numpy as np
 import tensorflow as tf
 
-from src.torch_compat.data_holder import DataHolder
+# from src.torch_compat.data_holder import DataHolder
 
 tf.get_logger().setLevel('DEBUG')
 
@@ -90,7 +90,6 @@ def __init__(self, config, models, config_path):
             self.malicious_clients[malicious_indices] = True
 
         self.global_dataset = self.build_dataset()
-        DataHolder.global_dataset = self.global_dataset
         # self.malicious_clients[np.random.choice(self.num_clients, self.num_malicious_clients, replace=False)] = True
         self.client_objs = []
         self.client_model = None

diff --git a/train_configs/basic/test_model_anticipate.yml b/train_configs/basic/test_model_anticipate.yml
@@ -41,7 +41,7 @@ client:
                 attacker_full_dataset: false
                 model_type: "test_model"
                 anticipate_steps: 3
-            name: AnticipateAttack
+            name: AnticipateTfAttack
         multi_attacker_scale_divide: true
     model_name: test_model
 dataset: