Coverage for maze_dataset/plotting/__init__.py: 100%

1"""utilities for plotting mazes and printing tokens 

2 

3- any `LatticeMaze` or `SolvedMaze` comes with a `as_pixels()` method that returns 

4 a 2D numpy array of pixel values, but this is somewhat limited 

5- `MazePlot` is a class that can be used to plot mazes and paths in a more customizable way 

6- `print_tokens` contains utilities for printing tokens, colored by their type, position, or some custom weights (i.e. attention weights) 

7""" 

8 

9from maze_dataset.plotting.plot_dataset import plot_dataset_mazes, print_dataset_mazes 

10from maze_dataset.plotting.plot_maze import DEFAULT_FORMATS, MazePlot, PathFormat 

11from maze_dataset.plotting.print_tokens import ( 

12 color_maze_tokens_AOTP, 

13 color_tokens_cmap, 

14 color_tokens_rgb, 

15) 

16 

17__all__ = [ 

18 # submodules 

19 "plot_dataset", 

20 "plot_maze", 

21 "plot_svg_fancy", 

22 "plot_tokens", 

23 "print_tokens", 

24 # imports 

25 "plot_dataset_mazes", 

26 "print_dataset_mazes", 

27 "DEFAULT_FORMATS", 

28 "MazePlot", 

29 "PathFormat", 

30 "color_tokens_cmap", 

31 "color_maze_tokens_AOTP", 

32 "color_tokens_rgb", 

33]