cosmetic fix

Author: mishadr

src/attacks/evasion_attacks.py

@@ -309,7 +309,7 @@ def attack(
             self,
             model_manager: Type,
             gen_dataset: GeneralDataset,
-            mask_tensor: torch.Tensor,
+            mask_tensor: torch.Tensor,  # is not used
             task_type: str = None,
     ) -> None:
         if task_type is None:

web_interface/back_front/attack_defense_blocks.py

@@ -200,8 +200,12 @@ def do(
 
             # Apply attacks
             metrics_values = {}
-            metrics_values['After attacks'] = self.model_manager.evaluate_model(
+            res = self.model_manager.evaluate_model(
                 gen_dataset=self.gen_dataset, metrics=self.metrics)
+            if self.ad_configs["AD-ea"] is not None:
+                metrics_values['After evasion attack'] = res
+            else:
+                metrics_values = res
 
             # Get MI metrics
             import numpy as np