This allows focussed experiments, e.g.

```python
from bsuite import sweep

exploration_sweep = sweep.TAGS['exploration']
```

Also add better type annotations to sweep & summary_analysis.

PiperOrigin-RevId: 306402609
Change-Id: I8b5fa0c0241360169e7ca8b8695a6e6f4849f9c1