fix: fix return type hint for grid(), add a few others

[jameslamb]

Contributes to #87.

Fixes the return type hint for grid(), which fixes this error reported by mypy:

src/rapids_dependency_file_generator/_rapids_dependency_file_generator.py:71: error: "Generator" expects 3 type arguments, but 1 given  [type-arg]

Also adds a few return type hints for places that were missing them, to improve the ability of type checkers to inspect the code.

Notes for Reviewers

What is this [..., None, None] in the Generator hint?

From the typing docs:

A generator can be annotated by the generic type Generator[YieldType, SendType, ReturnType].

If your generator will only yield values, set the SendType and ReturnType to None:

def infinite_stream(start: int) -> Generator[int, None, None]:
...

(docs link)

[GPUtester]

🎉 This PR is included in version 1.13.10 🎉

The release is available on:

• GitHub release
• PyPI release
• Conda release - rapidsai
• Conda release - rapidsai-nightly

Your semantic-release bot 📦🚀