Coverage for transformer_lens/model_bridge/bridge.py: 100%

1"""Compatibility re-export shim. 

2 

3Historically, :class:`TransformerBridge` lived in this module. The class was 

4split into three (``BridgeCore`` framework-agnostic parent, ``TransformerBridge`` 

5torch-backed, ``RemoteBridge`` non-torch) across their own files; this shim 

6preserves the original import path: 

7 

8 from transformer_lens.model_bridge.bridge import TransformerBridge 

9""" 

10from transformer_lens.model_bridge.bridge_core import BridgeCore 

11from transformer_lens.model_bridge.remote_bridge import RemoteBridge 

12from transformer_lens.model_bridge.transformer_bridge import TransformerBridge 

13 

14__all__ = ["BridgeCore", "RemoteBridge", "TransformerBridge"]