fix other test from rebase

jbloomAus · Oct 10, 2024 · 81a7805 · 81a7805
1 parent 3b78f10
commit 81a7805
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/unit/training/test_cache_activations_runner.py b/tests/unit/training/test_cache_activations_runner.py
@@ -192,6 +192,10 @@ def to_tokens(self, *args: Tuple[Any, ...], **kwargs: Any) -> torch.Tensor:
         def W_E(self) -> torch.Tensor:
             return torch.ones(16, 16)
 
+        @property
+        def cfg(self) -> LanguageModelSAERunnerConfig:
+            return cfg
+
     dataset = Dataset.from_list(
         [
             {"text": "hello world1"},