Coverage for transformer_lens/__init__.py: 88%

1from . import ( 

2 components, 

3 conversion_utils, 

4 evals, 

5 factories, 

6 head_detector, 

7 hook_points, 

8 patching, 

9 tools, 

10 train, 

11 utilities, 

12) 

13from . import loading_from_pretrained as loading 

14from . import supported_models 

15from .ActivationCache import ActivationCache 

16from .BertNextSentencePrediction import BertNextSentencePrediction 

17from .cache.key_value_cache import TransformerLensKeyValueCache 

18from .cache.key_value_cache_entry import TransformerLensKeyValueCacheEntry 

19from .config import HookedTransformerConfig, TransformerBridgeConfig 

20from .FactoredMatrix import FactoredMatrix 

21from .HookedEncoder import HookedEncoder 

22from .HookedAudioEncoder import HookedAudioEncoder 

23from .HookedEncoderDecoder import HookedEncoderDecoder 

24from .HookedTransformer import HookedTransformer 

25from .HookedRootModule import HookedRootModule 

26 

27# LIT integration (optional, requires lit-nlp package) 

28try: 

29 from . import lit 

30except ImportError: 

31 # LIT is an optional dependency 

32 lit = None # type: ignore 

33 

34from .SVDInterpreter import SVDInterpreter 

35 

36import os as _os # noqa: E402 

37 

38if _os.environ.get("TRANSFORMERLENS_HF_RETRY") == "1": 38 ↛ 43line 38 didn't jump to line 43 because the condition on line 38 was always true

39 from .utilities.hf_utils import enable_hf_retry as _enable_hf_retry # noqa: E402 

40 

41 _enable_hf_retry() 

42 

43__all__ = [ 

44 "HookedTransformerConfig", 

45 "TransformerBridgeConfig", 

46 "FactoredMatrix", 

47 "ActivationCache", 

48 "HookedTransformer", 

49 "SVDInterpreter", 

50 "HookedEncoder", 

51 "HookedEncoderDecoder", 

52 "HookedRootModule", 

53 "TransformerLensKeyValueCache", 

54 "TransformerLensKeyValueCacheEntry", 

55 "components", 

56 "conversion_utils", 

57 "factories", 

58 "utilities", 

59 "tools", 

60]