This turns out to be tricky to type with union args, see e.g. https://github.com/python/mypy/issues/18321