chore: add mypy and simplify Dataset.features logic #52

Closed ashwin153 closed 3 days ago