Coverage for transformer_lens/conversion_utils/conversion_steps/callable_tensor_conversion.py: 100%

1from collections.abc import Callable 

2 

3from .base_tensor_conversion import BaseTensorConversion 

4 

5 

6class CallableTensorConversion(BaseTensorConversion): 

7 def __init__(self, convert_callable: Callable): 

8 super().__init__() 

9 self.convert_callable = convert_callable 

10 

11 def handle_conversion(self, input_value: dict, *full_context): 

12 return self.convert_callable(input_value, *full_context) 

13 

14 def __repr__(self): 

15 return f"Is the following callable operation: {self.convert_callable}"