Add type hints to `trace` decorator.