Internal Type Inference

[Kludex]

Hi @charliermarsh 👋

I'm building bump-pydantic, and I was wondering if you can help me.

You don't do any kind of type inference on ruff, right?

I want to refactor the following code:

from pydantic import BaseModel

class Model(BaseModel):
    a: int

model = Model(a=1)
model.dict()  # change this to `model.model_dump()`

I want to make sure that model is a BaseModel to perform the refactor.

Do you do something like that in ruff?

[charliermarsh]

Yeah, we don't do this in Ruff right now. Our semantic model has enough information that we could support a rewrite in some cases like the one above, but there's a sliding scale of complexity, e.g. this is harder:

def f(x: int) -> Any:
  return Model(a=1)

model = f()
model.dict()

This is also harder:

from other_module import Model

model = Model(a=1)
model.dict()

[Kludex]

Yeah, I thought so.

Thanks anyway. 🙏

[charliermarsh]

You could look at using astroid which has some support for static inference.