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 train, 

10 utilities, 

11) 

12from . import loading_from_pretrained as loading 

13from . import supported_models 

14from .ActivationCache import ActivationCache 

15from .BertNextSentencePrediction import BertNextSentencePrediction 

16from .cache.key_value_cache import TransformerLensKeyValueCache 

17from .cache.key_value_cache_entry import TransformerLensKeyValueCacheEntry 

18from .config import HookedTransformerConfig, TransformerBridgeConfig 

19from .FactoredMatrix import FactoredMatrix 

20from .HookedEncoder import HookedEncoder 

21from .HookedAudioEncoder import HookedAudioEncoder 

22from .HookedEncoderDecoder import HookedEncoderDecoder 

23from .HookedTransformer import HookedTransformer 

24from .HookedRootModule import HookedRootModule 

25 

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

27try: 

28 from . import lit 

29except ImportError: 

30 # LIT is an optional dependency 

31 lit = None # type: ignore 

32 

33from .SVDInterpreter import SVDInterpreter 

34 

35import os as _os # noqa: E402 

36 

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

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

39 

40 _enable_hf_retry() 

41 

42__all__ = [ 

43 "HookedTransformerConfig", 

44 "TransformerBridgeConfig", 

45 "FactoredMatrix", 

46 "ActivationCache", 

47 "HookedTransformer", 

48 "SVDInterpreter", 

49 "HookedEncoder", 

50 "HookedEncoderDecoder", 

51 "HookedRootModule", 

52 "TransformerLensKeyValueCache", 

53 "TransformerLensKeyValueCacheEntry", 

54 "components", 

55 "conversion_utils", 

56 "factories", 

57 "utilities", 

58]