order_by method incorrectly infers return type of select

[SergeyKapshuchenko]

First Check

• I added a very descriptive title here.
• I used the GitHub search to find a similar question and didn't find it.
• I searched the SQLModel documentation, with the integrated search.
• I already searched in Google "How to X in SQLModel" and didn't find any information.
• I already read and followed all the tutorial in the docs and didn't find an answer.
• I already checked if it is not related to SQLModel but to Pydantic.
• I already checked if it is not related to SQLModel but to SQLAlchemy.

Commit to Help

• I commit to help with one of those options 👆

Example Code

from sqlmodel import Field, SQLModel, select


class Hero(SQLModel, table=True):
    id: int | None = Field(default=None, primary_key=True)
    age: int
    age_optional: int | None = Field(default=None)
    name: str


if __name__ == "__main__":
    query_1 = select(Hero).order_by(
        Hero.age
    )  # return type is Any if the field is required integer

    query_2 = select(Hero).order_by(
        Hero.age_optional
    )  # return type is SelectOfScalar[Hero] if the field is optional integer

Description

When using a required integer field with order_by in SQLModel, the inferred return type is 'Any'

Operating System

macOS

Operating System Details

No response

SQLModel Version

0.0.16

Python Version

3.12

Additional Context

No response

[SergeyKapshuchenko]

any updates ?

[YuriiMotov]

Works fine in current version (0.0.24) - both types are SelectOfScalar[Hero]