fix rename problem

ispras · Jul 23, 2024 · 725783f · 725783f
1 parent 8854f69
commit 725783f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/defense/evasion_defense.py b/src/defense/evasion_defense.py
@@ -7,10 +7,10 @@ class EvasionDefender(Defender):
     def __init__(self, **kwargs):
         super().__init__()
 
-    def pre_epoch(self, **kwargs):
+    def pre_batch(self, **kwargs):
         pass
 
-    def post_epoch(self, **kwargs):
+    def post_batch(self, **kwargs):
         pass
 
 

diff --git a/src/defense/mi_defense.py b/src/defense/mi_defense.py
@@ -5,18 +5,18 @@ class MIDefender(Defender):
     def __init__(self, **kwargs):
         super().__init__()
 
-    def pre_epoch(self, **kwargs):
+    def pre_batch(self, **kwargs):
         pass
 
-    def post_epoch(self, **kwargs):
+    def post_batch(self, **kwargs):
         pass
 
 
 class EmptyMIDefender(MIDefender):
     name = "EmptyMIDefender"
 
-    def pre_epoch(self, **kwargs):
+    def pre_batch(self, **kwargs):
         pass
 
-    def post_epoch(self, **kwargs):
+    def post_batch(self, **kwargs):
         pass