removed unnecessary args

Author: jkminder

dictionary_learning/trainers/batch_top_k.py

@@ -30,15 +30,13 @@ def __init__(
         seed: Optional[int] = None,
         device: Optional[str] = None,
         wandb_name: str = "BatchTopKSAE",
-        submodule_name: Optional[str] = None,
         activation_mean: Optional[t.Tensor] = None,
         activation_std: Optional[t.Tensor] = None,
     ):
         super().__init__(seed)
         assert layer is not None and lm_name is not None
         self.layer = layer
         self.lm_name = lm_name
-        self.submodule_name = submodule_name
         self.wandb_name = wandb_name
         self.steps = steps
         self.decay_start = decay_start
@@ -258,7 +256,6 @@ def config(self):
             "layer": self.layer,
             "lm_name": self.lm_name,
             "wandb_name": self.wandb_name,
-            "submodule_name": self.submodule_name,
         }
 
     @staticmethod