Think there's a bit of a conceptual mismatch here.

From the python typing.Generator docs:

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

Your current signature implies the returned Generator accepts a value via the generator.send method. This allows a caller to "send" values into generator functions, resuming execution.

However, the contextmanager protocol never sends values into generators -- instead only ever relying on next() to start/resume execution.

So the type should actually be:

Furthermore, we can simplify this type to an Iterator[YieldType] since the docs go on to suggest:

Alternatively, annotate your generator as having a return type of either Iterable[YieldType] or Iterator[YieldType]:

So since we only care about the YieldType (i.e InitializationType) we should simplify this typing to use the collections.abc.Iterator ABC:

Awesome 🙌 Thank you for sharing @gregfurman ✨

That's easiest to read despite PyCharm not liking it. I went ahead and replaced this and the other occurrence for get_environment

Maybe, PyCharm doesn't implement the with usage typings correctly here, yielding a warning 🤷‍♂️

The IDE and AI generate unnecesarily complex or wrong stuff here:

• IDE: Generator[Literal[InitializationType.provisioned_concurrency, InitializationType.on_demand], Any, None]
• Gemini 3: Generator[InitializationType, Any, None]
  • You correctly pointed out that Any should be None because the method does not have a return type
  • Question I asked it: "What are the correct typings for localstack.services.lambda_.invocation.counting_service.CountingService.get_invocation_lease?"

note: If we decide to go with my suggestion for Iterator this should be replaced:

Thanks for reminding to import from collections.abc and not the deprecated typings (first IDE suggestion)