[BUG] Type error calling init_beanie()

[markedwards]

Describe the bug
There is a type error calling init_beanie():

Argument "document_models" to "init_beanie" has incompatible type "list[type[Document]]"; expected "list[type[<nothing>] | type[View] | str] | None"  [arg-type]

To Reproduce

from beanie import Document, init_beanie
from motor.motor_asyncio import AsyncIOMotorClient


class MyModel(Document):
    foo: str


async def example() -> None:
    client = AsyncIOMotorClient("mongodb://user:pass@host:27017")
    await init_beanie(database=client.db_name, document_models=[MyModel])  # <- raises the type error

Expected behavior
No type error.

Additional context

[roman-right]

Hi @markedwards ,
Could you please tell what are you using? MyPy or pyright, and respective version. Thank you

[markedwards]

@roman-right -- mypy 1.5.1.

I believe you need to have strict=True set in order to reveal the error.

My guess is this can be resolved with this signature:

document_models: Optional[Sequence[Union[Type["DocType"], Type["View"], str]]]

or even simpler:

document_models: Optional[Sequence[Union[Type[Document], Type[View], str]]]

[cojua8]

I think you are right, that change should be enough