# LeapText

Used to visualize Text. Optionally, together with a heatmap that signifies words importance

```python
import numpy.typing as npt
from code_loader.contract.enums import LeapDataType

@dataclass
class LeapText:
    data: List[str]
    type: LeapDataType = LeapDataType.Text
    heatmap: Optional[List[float]] = None
```

<table><thead><tr><th width="167.39065467110788">Args</th><th></th></tr></thead><tbody><tr><td><code>data</code></td><td>a list of strings that compose the text (length N)</td></tr><tr><td><code>heatmap</code></td><td>(optional) - a lists of floats that provides word importance (length N)</td></tr></tbody></table>

## Examples

#### Basic Usage

```python
import numpy as np
from code_loader.contract.visualizer_classes import LeapText
from code_loader.contract.enums import LeapDataType
...

@tensorleap_custom_visualizer(name="input_visualizer", visualizer_type=LeapDataType.Text)
def input_visualizer(input_ids: np.ndarray) -> LeapText:
    input_ids = np.squeeze(input_ids)
    text = decode_token_ids(input_ids)
    return LeapText(text)
```