Coverage for transformer_lens/tools/__init__.py: 75%

1"""TransformerLens tools package. 

2 

3This package contains utilities and tools for working with TransformerLens, 

4including the model registry for discovering compatible HuggingFace models. 

5 

6Subpackages: 

7 - analysis: High-level interpretability analyses (e.g. Direct Logit Attribution) 

8 - model_registry: Tools for discovering and documenting supported models 

9 

10Subpackages load lazily (PEP 562): ``analysis`` pulls in torch-heavy core 

11modules, so an eager import here would create a cycle for core modules that 

12consume ``model_registry`` data at import time. 

13""" 

14 

15import importlib 

16from typing import Any 

17 

18_SUBMODULES = ("analysis", "model_registry", "training") 

19 

20__all__ = ["analysis", "model_registry", "training"] 

21 

22 

23def __getattr__(name: str) -> Any: 

24 if name in _SUBMODULES: 24 ↛ 25line 24 didn't jump to line 25 because the condition on line 24 was never true

25 return importlib.import_module(f".{name}", __name__) 

26 raise AttributeError(f"module {__name__!r} has no attribute {name!r}") 

27 

28 

29def __dir__() -> list[str]: 

30 return sorted(set(globals()) | set(_SUBMODULES))